DocUNet: Document Image Unwarping via A Stacked U-Net, Ke Ma, Zhixin Shu, Xue Bai, Jue Wang, Dimitris Samaras. Proceedings of IEEE Conference on Computer Vision and Pattern Recognition (CVPR)

数据生成

• 进入data目录，在config.json中填写配置文件，然后使用命令ulimit -s检查用户进程的堆栈软限制大小，Ubuntu默认为8Mb，建议调大，直到不报段错误，使用命令ulimit -s [size]，需要先调大硬限制
• 运行文件generate.py开始生成数据

图片反变形

• 导入inverse_deform模块
• 使用函数inverse_deform(img_path, label_path, save_path)即可反变形