English

UVDoc: Neural Grid-based Document Unwarping

Computer Vision and Pattern Recognition 2024-02-28 v2 Graphics

Abstract

Restoring the original, flat appearance of a printed document from casual photographs of bent and wrinkled pages is a common everyday problem. In this paper we propose a novel method for grid-based single-image document unwarping. Our method performs geometric distortion correction via a fully convolutional deep neural network that learns to predict the 3D grid mesh of the document and the corresponding 2D unwarping grid in a dual-task fashion, implicitly encoding the coupling between the shape of a 3D piece of paper and its 2D image. In order to allow unwarping models to train on data that is more realistic in appearance than the commonly used synthetic Doc3D dataset, we create and publish our own dataset, called UVDoc, which combines pseudo-photorealistic document images with physically accurate 3D shape and unwarping function annotations. Our dataset is labeled with all the information necessary to train our unwarping network, without having to engineer separate loss functions that can deal with the lack of ground-truth typically found in document in the wild datasets. We perform an in-depth evaluation that demonstrates that with the inclusion of our novel pseudo-photorealistic dataset, our relatively small network architecture achieves state-of-the-art results on the DocUNet benchmark. We show that the pseudo-photorealistic nature of our UVDoc dataset allows for new and better evaluation methods, such as lighting-corrected MS-SSIM. We provide a novel benchmark dataset that facilitates such evaluations, and propose a metric that quantifies line straightness after unwarping. Our code, results and UVDoc dataset are available at https://github.com/tanguymagne/UVDoc.

Keywords

Cite

@article{arxiv.2302.02887,
  title  = {UVDoc: Neural Grid-based Document Unwarping},
  author = {Floor Verhoeven and Tanguy Magne and Olga Sorkine-Hornung},
  journal= {arXiv preprint arXiv:2302.02887},
  year   = {2024}
}

Comments

14 pages, published in SIGGRAPH Asia 2023 Conference Papers

R2 v1 2026-06-28T08:33:10.329Z