Residential College | false |
Status | 已發表Published |
UWFormer: Underwater Image Enhancement via a Semi-Supervised Multi-Scale Transformer | |
Weiwen Chen1; Yingtie Lei1; Shenghong Luo1; Ziyang Zhou2; Mingxian Li2; Chi-Man Pun1 | |
2024-09 | |
Conference Name | 2024 International Joint Conference on Neural Networks (IJCNN) |
Source Publication | Proceedings of the International Joint Conference on Neural Networks (IJCNN) |
Conference Date | 30 June 2024 - 05 July 2024 |
Conference Place | Yokohama, Japan |
Country | Japan |
Publisher | Institute of Electrical and Electronics Engineers Inc. |
Abstract | Underwater images often exhibit poor quality, distorted color balance and low contrast due to the complex and intricate interplay of light, water, and objects. Despite the significant contributions of previous underwater enhancement techniques, there exist several problems that demand further improvement: (i) The current deep learning methods rely on Convolutional Neural Networks (CNNs) that lack the multi-scale enhancement, and global perception field is also limited. (ii) The scarcity of paired real-world underwater datasets poses a significant challenge, and the utilization of synthetic image pairs could lead to overfitting. To address the aforementioned problems, this paper introduces a Multi-scale Transformer-based Network called UWFormer for enhancing images at multiple frequencies via semi-supervised learning, in which we propose a Nonlinear Frequency-aware Attention mechanism and a Multi-Scale Fusion Feed-forward Network for low-frequency enhancement. Besides, we introduce a special underwater semi-supervised training strategy, where we propose a Subaqueous Perceptual Loss function to generate reliable pseudo labels. Experiments using full-reference and non-reference underwater benchmarks demonstrate that our method outperforms state-of-the-art methods in terms of both quantity and visual quality. The code is available at https://github.com/leiyingtie/UWFormer. |
Keyword | Semi-supervised Learning Underwater Image Enhancement Vision Transformer |
DOI | 10.1109/IJCNN60899.2024.10651366 |
Language | 英語English |
Scopus ID | 2-s2.0-85205005730 |
Fulltext Access | |
Citation statistics | |
Document Type | Conference paper |
Collection | Faculty of Science and Technology DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE |
Corresponding Author | Chi-Man Pun |
Affiliation | 1.University of Macau, Macau, Macao 2.Huizhou University, Huizhou, China |
First Author Affilication | University of Macau |
Corresponding Author Affilication | University of Macau |
Recommended Citation GB/T 7714 | Weiwen Chen,Yingtie Lei,Shenghong Luo,et al. UWFormer: Underwater Image Enhancement via a Semi-Supervised Multi-Scale Transformer[C]:Institute of Electrical and Electronics Engineers Inc., 2024. |
APA | Weiwen Chen., Yingtie Lei., Shenghong Luo., Ziyang Zhou., Mingxian Li., & Chi-Man Pun (2024). UWFormer: Underwater Image Enhancement via a Semi-Supervised Multi-Scale Transformer. Proceedings of the International Joint Conference on Neural Networks (IJCNN). |
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