Image restoration for real-world under-display imaging

KeMing Gao,Meng Chang,Kunjun Jiang,Yaxu Wang,Zhihai Xu,Huajun Feng,Qi Li,Zengxin Hu,YueTing Chen
DOI: https://doi.org/10.1364/OE.441256
IF: 3.8
2021-01-01
Optics Express
Abstract:Under-display imaging technique was recently proposed to enlarge the screen-to-body ratio for full-screen devices. However, existing image restoration algorithms have difficulty generalizing to real-world under-display (UD) images, especially to images containing strong light sources. To address this issue, we propose a novel method for building a synthetic dataset (CalibPSF dataset) and introduce a two-stage neural network to solve the under-display imaging degradation problem. The CalibPSF dataset is generated using the calibrated high dynamic range point spread function (PSF) of the under-display optical system and contains various simulated light sources. The two-stage network solves the color distortion and diffraction degradation in order. We evaluate the performance of our algorithm on our captured real-world test set. Comprehensive experiments demonstrate the superiority of our method in different dynamic range scenes. (C) 2021 Optical Society of America under the terms of the OSA Open Access Publishing Agreement.
What problem does this paper attempt to address?