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