YOLOv7-CHS: An Emerging Model for Underwater Object Detection
-
Published:2023-10-09
Issue:10
Volume:11
Page:1949
-
ISSN:2077-1312
-
Container-title:Journal of Marine Science and Engineering
-
language:en
-
Short-container-title:JMSE
Author:
Zhao Liang1ORCID, Yun Qing1, Yuan Fucai2, Ren Xu1, Jin Junwei3, Zhu Xianchao3
Affiliation:
1. College of Electrical Engineering, Henan University of Technology, Zhengzhou 450001, China 2. School of Mechanical and Electrical Engineering, Henan University of Technology, Zhengzhou 450001, China 3. School of Artificial Intelligence and Big Data, Henan University of Technology, Zhengzhou 450001, China
Abstract
Underwater target detection plays a crucial role in marine environmental monitoring and early warning systems. It involves utilizing optical images acquired from underwater imaging devices to locate and identify aquatic organisms in challenging environments. However, the color deviation and low illumination in these images, caused by harsh working conditions, pose significant challenges to an effective target detection. Moreover, the detection of numerous small or tiny aquatic targets becomes even more demanding, considering the limited storage and computing power of detection devices. To address these problems, we propose the YOLOv7-CHS model for underwater target detection, which introduces several innovative approaches. Firstly, we replace efficient layer aggregation networks (ELAN) with the high-order spatial interaction (HOSI) module as the backbone of the model. This change reduces the model size while preserving accuracy. Secondly, we integrate the contextual transformer (CT) module into the head of the model, which combines static and dynamic contextual representations to effectively improve the model’s ability to detect small targets. Lastly, we incorporate the simple parameter-free attention (SPFA) module at the head of the detection network, implementing a combined channel-domain and spatial-domain attention mechanism. This integration significantly improves the representation capabilities of the network. To validate the implications of our model, we conduct a series of experiments. The results demonstrate that our proposed model achieves higher mean average precision (mAP) values on the Starfish and DUO datasets compared to the original YOLOv7, with improvements of 4.5% and 4.2%, respectively. Additionally, our model achieves a real-time detection speed of 32 frames per second (FPS). Furthermore, the floating point operations (FLOPs) of our model are 62.9 G smaller than those of YOLOv7, facilitating the deployment of the model. Its innovative design and experimental results highlight its effectiveness in addressing the challenges associated with underwater object detection.
Funder
National Nature Science Foundation of China Science and Technology Research Project of Henan Province
Subject
Ocean Engineering,Water Science and Technology,Civil and Structural Engineering
Reference45 articles.
1. Wang, F., Zhu, J., Chen, L., Zuo, Y., Hu, X., and Yang, Y. (2020). Autonomous and In Situ Ocean Environmental Monitoring on Optofluidic Platform. Micromachines, 11. 2. Qi, S., Du, J.F., Wu, M., Yi, H., Tang, L., Qian, T., and Wang, X. (2022, January 23–27). Underwater Small Target Detection Based on Deformable Convolutional Pyramid. Proceedings of the ICASSP 2022—2022 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), Singapore. 3. Yuan, X., Guo, L., Luo, C., Zhou, X., and Yu, C. (2022). A Survey of Target Detection and Recognition Methods in Underwater Turbid Areas. Appl. Sci., 12. 4. A comprehensive review of object detection with deep learning;Kaur;Digit. Signal Process.,2022 5. Underwater object detection algorithm based on feature enhancement and progressive dynamic aggregation strategy;Hua;Pattern Recognit.,2023
|
|