UM  > Faculty of Science and Technology
Residential Collegefalse
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 Name2024 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)
Source PublicationProceedings of the IEEE Computer Society Conference on Computer Vision and Pattern Recognition
Pages2900-2909
Conference Date16-22 June 2024
Conference PlaceSeattle, WA, USA
CountryUSA
PublisherIEEE 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.

KeywordComputer Vision Shape Computational Modeling Noise Reduction Training Data Boosting Data Models Pre-trained Models Image Restoration Spatial-varying Enhancement Channel-spatial Attention
DOI10.1109/CVPR52733.2024.00280
URLView the original
Language英語English
Scopus ID2-s2.0-85203166578
Fulltext Access
Citation statistics
Document TypeConference paper
CollectionFaculty of Science and Technology
INSTITUTE OF COLLABORATIVE INNOVATION
DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE
Corresponding AuthorLiu, Zhe
Affiliation1.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.
Related Services
Recommend this item
Bookmark
Usage statistics
Export to Endnote
Google Scholar
Similar articles in Google Scholar
[Xu, Xiaogang]'s Articles
[Kong, Shu]'s Articles
[Hu, Tao]'s Articles
Baidu academic
Similar articles in Baidu academic
[Xu, Xiaogang]'s Articles
[Kong, Shu]'s Articles
[Hu, Tao]'s Articles
Bing Scholar
Similar articles in Bing Scholar
[Xu, Xiaogang]'s Articles
[Kong, Shu]'s Articles
[Hu, Tao]'s Articles
Terms of Use
No data!
Social Bookmark/Share
All comments (0)
No comment.
 

Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.