HumanEric/docUnet.pytorch

This is a pytorch implementation of DocUNet: Document Image Unwarping via A Stacked U-Net

DocUNet: Document Image Unwarping via A Stacked U-Net

paper zhihu

how to run

1. prepare some scans and background images
2. use flow scipt to generated some images and labels

python3 data_generator/generator.py -i scan_images_path -b background_images_path -o output_path

3. change trainroot in config.py
4. use flow scipt to train

python3 train.py

how to eval

1. change model_path and img_path in predict.py
2. use flow scipt to train

python3 predict.py

generated images sample

result

result on training data

left：input, right output

result on really data

left：output, right input