/DocUNet

An unofficial implementation of DocUNet

Primary LanguagePython

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)即可反变形

训练结果