├── .gitignore ├── README.md ├── constants.py ├── core ├── models │ ├── deeplabv3_plus.py │ ├── mobilenet_v2_dilation.py │ ├── mobilenetv2.py │ ├── pspnet.py │ ├── resnet.py │ ├── unet.py │ ├── unet_paper.py │ └── unet_pytorch.py └── trainers │ └── trainer.py ├── dataloader ├── docunet.py ├── docunet_im2im.py └── docunet_inverted.py ├── environment.yml ├── main.py ├── parser_options.py ├── playground.py ├── readme_images ├── generating_deformed_images.PNG ├── output_examples.png └── overall_architecture.PNG ├── train.sh └── util ├── custom_transforms.py ├── general_functions.py ├── losses.py ├── lr_scheduler.py ├── ms_ssim.py ├── ssim.py └── summary.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/constants.py -------------------------------------------------------------------------------- /core/models/deeplabv3_plus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/core/models/deeplabv3_plus.py -------------------------------------------------------------------------------- /core/models/mobilenet_v2_dilation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/core/models/mobilenet_v2_dilation.py -------------------------------------------------------------------------------- /core/models/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/core/models/mobilenetv2.py -------------------------------------------------------------------------------- /core/models/pspnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/core/models/pspnet.py -------------------------------------------------------------------------------- /core/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/core/models/resnet.py -------------------------------------------------------------------------------- /core/models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/core/models/unet.py -------------------------------------------------------------------------------- /core/models/unet_paper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/core/models/unet_paper.py -------------------------------------------------------------------------------- /core/models/unet_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/core/models/unet_pytorch.py -------------------------------------------------------------------------------- /core/trainers/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/core/trainers/trainer.py -------------------------------------------------------------------------------- /dataloader/docunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/dataloader/docunet.py -------------------------------------------------------------------------------- /dataloader/docunet_im2im.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/dataloader/docunet_im2im.py -------------------------------------------------------------------------------- /dataloader/docunet_inverted.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/dataloader/docunet_inverted.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/environment.yml -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/main.py -------------------------------------------------------------------------------- /parser_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/parser_options.py -------------------------------------------------------------------------------- /playground.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/playground.py -------------------------------------------------------------------------------- /readme_images/generating_deformed_images.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/readme_images/generating_deformed_images.PNG -------------------------------------------------------------------------------- /readme_images/output_examples.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/readme_images/output_examples.png -------------------------------------------------------------------------------- /readme_images/overall_architecture.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/readme_images/overall_architecture.PNG -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/train.sh -------------------------------------------------------------------------------- /util/custom_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/util/custom_transforms.py -------------------------------------------------------------------------------- /util/general_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/util/general_functions.py -------------------------------------------------------------------------------- /util/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/util/losses.py -------------------------------------------------------------------------------- /util/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/util/lr_scheduler.py -------------------------------------------------------------------------------- /util/ms_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/util/ms_ssim.py -------------------------------------------------------------------------------- /util/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/util/ssim.py -------------------------------------------------------------------------------- /util/summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mhashas/Document-Image-Unwarping-pytorch/HEAD/util/summary.py --------------------------------------------------------------------------------