DocEnTr: An End-to-End Document Image Enhancement Transformer

Document images can be affected by many degradation scenarios, which cause recognition and processing difficulties. In this age of digitization, it is important to denoise them for proper usage. To address this challenge, we present a new encoder-decoder architecture based on vision transformers to enhance both machine-printed and handwritten document images, in an end-to-end fashion. The encoder operates directly on the pixel patches with their positional information without the use of any convolutional layers, while the decoder reconstructs a clean image from the encoded patches. Conducted experiments show a superiority of the proposed model compared to the state-of the-art methods on several DIBCO benchmarks. Code and models will be publicly available at: \url{https://github.com/dali92002/DocEnTR}.

PDF Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Binarization DIBCO 2011 DocEnTr-Base{8} PSNR 20.81 # 2
F-Measure 94.37 # 5
FPS 96.15 # 2
DRD 1.63 # 3
Binarization DIBCO 2017 DocEnTr-Base{8} F-Measure 92.53 # 3
FPS 95.15 # 1
DRD 2.37 # 2
PSNR 19.11 # 3
Binarization H-DIBCO 2011 DocEnTr-Base{8} PNSR 22.29 # 1
F-Measure 95.31 # 1
FPS 96.29 # 1
DRD 1.6 # 1
Binarization H-DIBCO 2018 DocEnTr-Base{8} PSNR 19.46 # 6
F-Measure 90.59 # 6
FPS 93.97 # 2
DRD 3.35 # 5

Methods


No methods listed for this paper. Add relevant methods here