/My-DocTr-Plus

Primary LanguagePythonMIT LicenseMIT

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.

this is first version model file:

链接：https://pan.baidu.com/s/16G3ZxPK_Zp8cbxrcM6YfSA 提取码：oklm

How to use

  1. Put the pretrained model to $ROOT/model_save/.
  2. Put the distorted images in $ROOT/distorted/.
  3. Run the script and the rectified images are saved in $ROOT/rectified/ by default. 
    python inference.py

some results of first version:

@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} }