this code just record my-self released first version result for unwarp distorted images and Referenced some codes from https://github.com/fh2019ustc/DocTr-Plus and thanks for sharing.
链接:https://pan.baidu.com/s/16G3ZxPK_Zp8cbxrcM6YfSA 提取码:oklm
- Put the pretrained model to
$ROOT/model_save/
. - Put the distorted images in
$ROOT/distorted/
. - Run the script and the rectified images are saved in
$ROOT/rectified/
by default.python inference.py
@inproceedings{feng2021doctr, title={DocTr: Document Image Transformer for Geometric Unwarping and Illumination Correction}, author={Feng, Hao and Wang, Yuechen and Zhou, Wengang and Deng, Jiajun and Li, Houqiang}, booktitle={Proceedings of the 29th ACM International Conference on Multimedia}, pages={273--281}, year={2021} }
@article{feng2023doctrp, title={Deep Unrestricted Document Image Rectification}, author={Feng, Hao and Liu, Shaokai and Deng, Jiajun and Zhou, Wengang and Li, Houqiang}, journal={arXiv}, year={2023} }