Real-World Blind Face Restoration with Generative Facial Prior and Degradation Simulation

Weihong Zhu,Changhui Hu,Lintao Xu
DOI: https://doi.org/10.1109/ICSIP61881.2024.10671462
2024-07-12
Abstract:Real-world blind face restoration is very hard, since real-world blind face images are with unknown complex multiple degradations. Our previous work [1] proposed a joint image-to-image translation (JI2IT) method for real-world blind traffic monitoring driver face (TMDF) image restoration, whereas JI2IT is unsatisfactory to restore the facial details of real-world blind TMDF images. To tackle above issue, this paper proposes a RestormGAN for real-world image (i.e., TMDF image) restoration which consists of a channel axis attention based Restormer (CAR) encoder and a generative facial prior (i.e., GAN prior). Specifically, we propose to insert a novel channel height and width axes attention (CHWA) module into Restormer module to form CAR module, where CHWA contains the height-axis attention module (HAM) and the width-axis attention module (WAM). The HAM learns a weight for the height-axis of each channel while WAM learns a weight for the width-axis of each channel. Then the dual down-sampling (DDS) module is proposed to ensure the CAR encoder can be efficiently connected with GAN prior. DDS uses bilinear sampling and pixelunshuffle in parallel to generate more representative parameters. Finally, we propose to combine multi-handcraft degradations strategy and pretrained degradation model to construct massive image pairs for the training of RestormGAN. Extensive experiments demonstrate that our RestormGAN achieves state-of-the-art performances for real-world face restoration and superior results in traffic monitoring scenarios.
Engineering,Computer Science
What problem does this paper attempt to address?