SCN: A Novel Underwater Images Enhancement Method Based on Single Channel Network Model

被引:0
作者
Zhou, Fuheng [1 ,2 ]
Zhang, Siqing [1 ,2 ]
Huang, Yulong [1 ,2 ]
Zhu, Pengsen [1 ,2 ]
Zhang, Yonggang [1 ,2 ]
机构
[1] Harbin Engn Univ, Coll Intelligent Syst Sci & Engn, Harbin 150001, Peoples R China
[2] Minist Educ, Engn Res Ctr Nav Instruments, Harbin 150001, Peoples R China
基金
中国国家自然科学基金;
关键词
Image color analysis; Attenuation; Data models; Feature extraction; Deep learning; Image enhancement; Transformers; Image restoration; Convolutional neural networks; Green products; Convolutional neural network (CNN); multiple scale fusion mechanism; single channel network (SCN) model; transformer; underwater images enhancement;
D O I
暂无
中图分类号
TU [建筑科学];
学科分类号
0813 ;
摘要
Light is absorbed, reflected, and refracted in an underwater environment due to the interaction between water and light. The red and blue channels in an image are attenuated due to these interactions. The red, green, and blue channels are typically employed as inputs for deep learning models, and the color casts, which result from different attenuation rates of the three channels, may affect the model's generalization performance. Besides, the color casts existing in the reference images will impact the deep-learning models. To address these challenges, a single channel network (SCN) model is introduced, which exclusively employs the green channel as its input, and is unaffected by the attenuations in the red and blue channels. An innovative feature processing module is presented, in which the characteristics of transformers and convolutional layers are fused to capture nonlinear relationships among the red, green, and blue channels. The public EUVP and LSUI data set experiments show that the proposed SCN model achieves competitive results with the existing best three channel models for the case of slight signal attenuation, and outperforms the existing state of arts three-channel models for the case of strong signal attenuation. Furthermore, the proposed model is trained on the self-built noncolor biased underwater image data set and is also tested on the public UCCS data set with three different types of color casts, whose experimental results exhibit balanced color distribution.
引用
收藏
页码:758 / 775
页数:18
相关论文
共 58 条
[1]   Facial Emotion Recognition Using Transfer Learning in the Deep CNN [J].
Akhand, M. A. H. ;
Roy, Shuvendu ;
Siddique, Nazmul ;
Kamal, Md Abdus Samad ;
Shimamura, Tetsuya .
ELECTRONICS, 2021, 10 (09)
[2]  
Almahairi A, 2018, PR MACH LEARN RES, V80
[3]  
Ancuti C, 2012, PROC CVPR IEEE, P81, DOI 10.1109/CVPR.2012.6247661
[4]   Deep underwater image enhancement through colour cast removal and optimization algorithm [J].
Azmi, Kamil Zakwan Mohd ;
Ghani, Ahmad Shahrizan Abdul ;
Yusof, Zulkifli Md ;
Ibrahim, Zuwairie .
IMAGING SCIENCE JOURNAL, 2019, 67 (06) :330-342
[5]  
Carlevaris-Bianco N, 2010, OCEANS-IEEE
[6]  
Chambah M, 2004, PROC SPIE, V5293, P157
[7]  
Chiang JY, 2011, LECT NOTES COMPUT SC, V6915, P372, DOI 10.1007/978-3-642-23687-7_34
[8]   Transmission Estimation in Underwater Single Images [J].
Drews-, P., Jr. ;
do Nascimento, E. ;
Moraes, F. ;
Botelho, S. ;
Campos, M. .
2013 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION WORKSHOPS (ICCVW), 2013, :825-830
[9]   Automatic Red-Channel underwater image restoration [J].
Galdran, Adrian ;
Pardo, David ;
Picon, Artzai ;
Alvarez-Gila, Aitor .
JOURNAL OF VISUAL COMMUNICATION AND IMAGE REPRESENTATION, 2015, 26 :132-145
[10]   Underwater image quality enhancement through composition of dual-intensity images and Rayleigh-stretching [J].
Ghani, Ahmad Shahrizan Abdul ;
Isa, Nor Ashidi Mat .
SPRINGERPLUS, 2014, 3 :1-14