search

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

Does the codes support GPU? #21

Open StevenSK-king opened 1 year ago

StevenSK-king commented 1 year ago

Thanks for your great work first. When testing, it seems the process is by using the CPU. Can you use the GPU power when run inference_ill.py?