Multi‐task image restoration network based on spatial aggregation attention and multi‐feature fusion
Abstract The main purpose of image restoration is to recover high‐quality image content from degraded versions. However, current mainstream models tend to focus solely on spatial details or contextual semantics, resulting in poor repair effects. To address this issue, a multi‐task image repair network based on spatial aggregation attention and multi‐feature fusion (SAAM) is proposed. It utilizes the global semantic information from the low‐resolution subnetwork to guide the local feature extraction of the high‐resolution subnetwork, thereby preserving the overall image structure while enhancing local details. Additionally, to enhance the model's understanding and representation capabilities of images, the feature fusion mechanism (FFM) is designed to merge feature information from different levels. Finally, the spatial aggregation attention mechanism SAAM enhances the accuracy and quality of image restoration by weighting the importance of different regions in the image at multiple scales. The experimental results demonstrate that the proposed SAAM method outperforms similar approaches in image denoising, deraining and decracking tasks in peak signal‐to‐noise ratio, structural similarity and learned perceptual image patch similarity metrics. The model also exhibits promising performance in restoring real old photos and murals which demonstrates its generalizability.
Authors
- Yangbo Chen (ORCID: https://orcid.org/0009-0002-0230-7121)
- Xueya Zhao (ORCID: https://orcid.org/0009-0006-7993-9863)
- Wanqing Zhang (ORCID: https://orcid.org/0000-0002-7298-5983)
- Chunyan Peng (ORCID: https://orcid.org/0009-0001-2674-1750)
- Yuhui Zheng
Institutions
- Nanjing University of Information Science and Technology (CN)
- Qinghai Normal University (CN)
Publication Details
- Journal
- IET Image Processing
- Published
- 2024-11-03
- DOI
- https://doi.org/10.1049/ipr2.13268
- Primary Topic
- Advanced Image Fusion Techniques
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China
- Qinghai Provincial Department of Science and Technology
- National Key Research and Development Program of China