fh2019ustc / DocTr

The official code for “DocTr: Document Image Transformer for Geometric Unwarping and Illumination Correction”, ACM MM, Oral Paper, 2021.
Other
345 stars 48 forks source link

pls add environment in README #5

Closed yellowjs0304 closed 2 years ago

yellowjs0304 commented 2 years ago

Hi, Thank you for sharing your work. I wanna try the inference.py, but I'm getting in stuck with setting. could you please share the environment or the requirements.txt? (CUDA Version, pytorch version, other packages .. etc)

fh2019ustc commented 2 years ago

Thanks for your advice. We will add the environment requirement.