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

训练数据例子

链接: https://pan.baidu.com/s/1gGjnuuC0nRodtVmzt4GOgg 提取码: pdeq

预模型

coming soon

其他

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