Residential College | false |
Status | 已發表Published |
Boosting Image Restoration via Priors from Pre-Trained Models | |
Xu, Xiaogang1,2,3; Kong, Shu5,6,7; Hu, Tao3,8; Liu, Zhe1; Bao, Hujun1,4 | |
2024-09 | |
Conference Name | 2024 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) |
Source Publication | Proceedings of the IEEE Computer Society Conference on Computer Vision and Pattern Recognition |
Pages | 2900-2909 |
Conference Date | 16-22 June 2024 |
Conference Place | Seattle, WA, USA |
Country | USA |
Publisher | IEEE Computer Society |
Abstract | Pre-trained models with large-scale training data, such as CLIP and Stable Diffusion, have demonstrated remarkable performance in various high-level computer vision tasks such as image understanding and generation from language descriptions. Yet, their potential for low-level tasks such as image restoration remains relatively unexplored. In this paper, we explore such models to enhance image restoration. As off-the-shelf features (OSF) from pre-trained models do not directly serve image restoration, we propose to learn an additional lightweight module called Pre-Train-Guided Refinement Module (PTG-RM) to refine restoration results of a target restoration network with OSF. PTG-RM consists of two components, Pre-Train-Guided Spatial-Varying Enhancement (PTG-SVE), and Pre-Train-Guided Channel-Spatial Attention (PTG-CSA). PTG-SVE enables optimal short- and long-range neural operations, while PTG-CSA enhances spatial-channel attention for restoration-related learning. Extensive experiments demonstrate that PTG-RM, with its compact size (<1M parameters), effectively enhances restoration performance of various models across different tasks, including low-light enhancement, deraining, deblurring, and denoising. |
Keyword | Computer Vision Shape Computational Modeling Noise Reduction Training Data Boosting Data Models Pre-trained Models Image Restoration Spatial-varying Enhancement Channel-spatial Attention |
DOI | 10.1109/CVPR52733.2024.00280 |
URL | View the original |
Language | 英語English |
Scopus ID | 2-s2.0-85203166578 |
Fulltext Access | |
Citation statistics | |
Document Type | Conference paper |
Collection | Faculty of Science and Technology INSTITUTE OF COLLABORATIVE INNOVATION DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE |
Corresponding Author | Liu, Zhe |
Affiliation | 1.Zhejiang Lab, China 2.Cuhk, Hong Kong 3.RealityEdge 4.Zhejiang University 5.University of Macau, Macao 6.Institute of Collaborative Innovation, Macao 7.Texas A&M University 8.National University of Singapore, Singapore |
Recommended Citation GB/T 7714 | Xu, Xiaogang,Kong, Shu,Hu, Tao,et al. Boosting Image Restoration via Priors from Pre-Trained Models[C]:IEEE Computer Society, 2024, 2900-2909. |
APA | Xu, Xiaogang., Kong, Shu., Hu, Tao., Liu, Zhe., & Bao, Hujun (2024). Boosting Image Restoration via Priors from Pre-Trained Models. Proceedings of the IEEE Computer Society Conference on Computer Vision and Pattern Recognition, 2900-2909. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment