← all work generative-vision-lab

Document shadow removal

Remove shadows cast across a photographed document page.

Result

The finding

Notes

- Fine-tuned for 56 epochs on 800 training pairs; the best val score came at epoch 50 and five further epochs did not beat it, so the run was stopped there. - Train 30.67 / val 30.49 / test 30.60 - a tight spread across all three splits. - Samples show the shadowed input, the published weights' output, this run's output and the ground truth.

Output

Same subject across the row; each column is the same stage in every project.

Elsewhere

Code and full results on GitHub

Every number is read from the file the run produced. github.com/hammasbuilds · kaggle.com/muhammadhammas13