/DocumentPhotoCorrection

Document photo perspective transformation correction,利用unet网络进行 文档照片透视纠偏

Primary LanguageC++

DocumentPhotoCorrection

Document photo perspective transformation correction,利用unet网络进行 文档照片透视纠偏

环境(Requirements)

pip install -r requirements.txt

例子🌰(Demo)

  • 修改inference 中main函数所需路径

python infer.py

训练(train)

  • 修改train.py 中checkpoint_path 为模型路径
  • 修改data.py 中training_data_path 为训练数据路径

python train.py

可视化实例

例子🌰1

原图 结果

例子🌰2

原图 结果

例子🌰3

原图 结果

训练数据例子

预模型

  • coming soon

其他

  • 代码还在优化中,效果与效率都在持续更新....
  • 更多OCR,文档解析,物体检测,跟踪等分享,尽在微信公众号 hulugeAI