基于改进 YOLOv3 算法的水下小目标分类与识别

doi:10.12066/j.issn.1007-2861.2279

Abstract

Abstract:

This study proposes an improved YOLOv3 algorithm designed to address the twin issues of low detection and recognition rate and high false alarm rate with respect to the detection of small targets by sonar. The improved YOLOv3 network is optimised on the basis of the original YOLOv3 algorithm, with the hierarchical connection of the network changed and the features of the shallow and deep layers fused to form a new larger-scale detection layer. Concurrently, the linear scaling $K$-means clustering algorithm is used to optimise the calculation of the number of a priori boxes and the aspect ratio, thereby improving the correlation between the a priori and ground truth boxes. These modifications improve the average accuracy of the YOLOv3 algorithm by 7%. Experimental results show that the proposed improvements to the YOLOv3 algorithm result in the effective identification of small targets with higher accuracy and lower false alarm rate, while maintaining the real-time processing capabilities of the YOLOv3 algorithm.

Key words: YOLOv3, small target detection, deep learning

CLC Number:

TB567

SHAO Huixiang, ZENG Dan. Classification and recognition of underwater small targets based on improved YOLOv3 algorithm[J]. Journal of Shanghai University（Natural Science Edition）, 2021, 27(3): 481-491.

Add to citation manager EndNote|Reference Manager|ProCite|BibTeX|RefWorks

URL: https://www.journal.shu.edu.cn/EN/10.12066/j.issn.1007-2861.2279

https://www.journal.shu.edu.cn/EN/Y2021/V27/I3/481

Figures/Tables 10

Fig. 1

Fig. 2

Fig. 3

Fig. 4

Table 1

Fig. 5

Table 2

Table 3

Table 4

Table 5

[1]	Sherin B M, Supriya M H. SOS based selection and parameter optimization for underwater target classification[C]// IEEE OCEANS 2016 MTS. 2016: 1-4.
[2]	Ren S, He K, Girshick R, et al. Faster R-CNN: towards real-time object detection with region proposal networks[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2017,39(6):1137-1149. doi: 10.1109/TPAMI.2016.2577031
[3]	Redmon J, Divvala S, Girshick R, et al. You only look once: unified, real-time object detection[C]// IEEE Conference on Computer Vision and Pattern Recognition. 2016.
[4]	Liu W, Anguelov D, Erhan D, et al. SSD: single shot multibox detector[C]// European Conference on Computer Vision. 2016.
[5]	Redmon J, Farhadi A. YOLOv3: an incremental improvement[C]// International Conference on Computer Vision and Pattern Recognition. 2017.
[6]	Li Z, Xu X, Shen F, et al. CR-FPN: channel relation feature pyramid network for object detection[J]. Wireless Networks, 2020, DOI: 10.1007/s11276-020-02391-3. doi: 10.1007/s11276-020-02391-3
[7]	Li J L, Luo S J, Zhu Z Q, et al. 3D IoU-Net: IoU guided 3D object detector for point clouds[EB/OL]. (2020-04-10)[2020-09-20]. https://arxiv.org/abs/2004.04962.
[8]	Li C L, Ravanbakhsh S, Poczos B. Annealing Gaussian into ReLU: a new sampling strategy for Leaky-ReLU RBM[EB/OL]. (2016-11-11)[2020-09-20]. https://arxiv.org/abs/1611.03879.
[9]	Xu L, Choy C S, Li Y W. Deep sparse rectifier neural networks for speech denoising[C]// 2016 IEEE International Workshop on Acoustic Signal Enhancement. 2016.
[10]	Rathi D, Jain S, Indu S. Underwater fish species classification using convolutional neural network and deep learning [C]// 2017 Ninth International Conference on Advances in Pattern Recognition (ICAPR). 2017.
[11]	Smith L N. Cyclical learning rates for training neural networks[C]// 2017 IEEE Winter Conference on Applications of Computer Vision. 2017.
[12]	Zhang F. Modular configuration of service elements based on the improved $K$-means algorithm[J]. Expert Systems, 2019,36(5):e12344.
[13]	Szegedy C, Ioffe S, Vanhoucke V, et al. Inception-v4, Inception-ResNet and the impact of residual connections on learning[C]// International Conference on Learning Representations. 2016.
[14]	Santurkar S, Tsipras D, Ilyas A, et al. How does batch normalization help optimization?[C]// 32nd Conference on Neural Information Processing Systems. 2018.
[15]	宋刚, 杜宏伟, 王平, 等. 纹理细节保持的图像插值算法[J]. 计算机科学, 2019(增刊 1):169-176.
	Song G, Du H W, Wang P, et al. Image interpolation algorithm for texture detail preservation[J]. Computer Science, 2019(Suppl.1):169-176.
[16]	秦玺淳. 数字图像处理技术 [J]. 数字通信世界, 2017(12): 174, 207.
	Qin X C. Digital image processing technology [J]. Digital Communication World, 2017(12): 174, 207.
[17]	赵永强, 饶元, 董世鹏, 等. 深度学习目标检测方法综述[J]. 中国图象图形学报, 2020,25(4):5-30.
	Zhao Y Q, Rao Y, Dong S P, et al. Overview of deep learning target detection methods[J]. Journal of Image and Graphics, 2020,25(4):5-30.

Classification and recognition of underwater small targets based on improved YOLOv3 algorithm

RichHTML

PDF

Knowledge

Abstract

Cite this article

share this article

Figures/Tables 10

Related Articles 15

Recommended Articles

Metrics

Comments

[1]	CHU Qingyuan, CHU Haijian. A deep learning-based non-destructive testing method by combining ultrasonic phased array sector scanning and total focusing method [J]. Journal of Shanghai University（Natural Science Edition）, 2026, 32(2): 240-250.
[2]	ZHU Shuo, WANG Yongfang, LI Zixuan, CHEN Wei. Real-time reflective vest detection in operational environments based on YOLO [J]. Journal of Shanghai University（Natural Science Edition）, 2025, 31(4): 691-703.
[3]	FAN Shenwei, LI Guoping, WANG Guozhong. Lightweight image compression algorithm based on deep learning [J]. Journal of Shanghai University（Natural Science Edition）, 2024, 30(3): 466-475.
[4]	ZHANG Heng, SHU Zhan. Parametric identiﬁcation method for fabricated components based on point cloud deep learning [J]. Journal of Shanghai University（Natural Science Edition）, 2023, 29(3): 502-.
[5]	LI Zhongyu, SUN Haodong, LI Jiao. Design of real-time basketball referee gesture recognition system based on loss weighting [J]. Journal of Shanghai University（Natural Science Edition）, 2023, 29(1): 68-81.
[6]	YANG Yingjie, WANG Yongfang, ZHANG Han. HDR video reconstruction based on attention and feedback mechanism [J]. Journal of Shanghai University（Natural Science Edition）, 2023, 29(1): 56-67.
[7]	LI Chengfan, ZHAO Junjuan. Improved approach to detect small sample target based on remote sensing image [J]. Journal of Shanghai University（Natural Science Edition）, 2022, 28(2): 314-323.
[8]	WANG Rongrong, XU Shugong, HUANG Jianbo. Image matting based on deep learning [J]. Journal of Shanghai University（Natural Science Edition）, 2022, 28(2): 261-269.
[9]	XU Genglin, RAN Feng, DENG Liang, SHI Huakang, GUO Aiying. Application of lightweight neural network and Hash tracking algorithm in embedded face capture system [J]. Journal of Shanghai University（Natural Science Edition）, 2021, 27(6): 1018-1028.
[10]	XING Yixue, ZHU Yonghua, GAO Haiyan, ZHOU Jin, ZHANG Ke. Distant supervision for relation extraction via attention CNNs [J]. Journal of Shanghai University（Natural Science Edition）, 2021, 27(5): 983-992.
[11]	CHEN Yu, DING Youdong, YU Bing, XU Min. Video colourisation based on voxel flow [J]. Journal of Shanghai University（Natural Science Edition）, 2021, 27(1): 18-27.
[12]	Jiacheng HU, Xiangyang WANG, Han LIU. Defect detection of continuous casting slabs based on deep learning [J]. Journal of Shanghai University（Natural Science Edition）, 2019, 25(4): 445-452.
[13]	XIE Zhifeng, YE Guanhua, YAN Shuqi, HE Shaorong, DING Youdong. HDR image style transfer technique based on generative adversarial networks [J]. Journal of Shanghai University（Natural Science Edition）, 2018, 24(4): 524-534.
[14]	LANG Yue1, ZHOU Jiting1, LIANG Xiaolong2, ZHANG Wenjun1. Automatic annotation for film and Television drama shots and recut system based on face identification [J]. Journal of Shanghai University（Natural Science Edition）, 2017, 23(3): 353-363.
[15]	SHEN Wenfeng, ZHANG Jianlei, ZHOU Dingqian, CHEN Shengbo, QIU Feng. Recognition of Chinese characters on license plates based on big data [J]. Journal of Shanghai University（Natural Science Edition）, 2016, 22(1): 88-96.