改进YOLOv8的水下目标检测  被引量:2

Underwater Target Detection via Improved YOLOv8

在线阅读下载全文

作  者:周昕 李远禄 吴明轩 范小婷 王键翔 ZHOU Xin;LI Yuan-Lu;WU Ming-Xuan;FAN Xiao-Ting;WANG Jian-Xiang(School of Automation,Nanjing University of Information Science&Technology,Nanjing 210044,China;Jiangsu Collaborative Innovation Center for Atmospheric Environment and Equipment Technology,Nanjing 210044,China)

机构地区:[1]南京信息工程大学自动化学院,南京210044 [2]江苏省大气环境与装备技术协同创新中心,南京210044

出  处:《计算机系统应用》2024年第11期177-185,共9页Computer Systems & Applications

基  金:国家自然科学基金(61671010);青蓝工程(2191091900101)。

摘  要:针对水下目标检测中的尺度不一、重叠遮挡目标的漏检问题,提出了一种改进的YOLOv8水下目标检测算法.首先,在主干网络中引入可变形卷积(deformable convolution network, DCN),通过卷积核自适应形变的机制,提高模型对重叠遮挡目标的特征提取能力;其次,设计了一种空洞卷积空间金字塔模块(atrous spatial pyramid faster, ASPF),扩大输出特征图的感受野,提高模型对水下多尺度目标的感知能力;最后,对损失函数进行改进,优化模型的训练过程并提高定位精度.将改进算法在URPC数据集上进行实验,结果表明改进算法的检测精度达到了87.3%,相较于原始算法YOLOv8提高了3.4%,同时能够精准检测水下多尺度、重叠遮挡目标.An improved YOLOv8 algorithm for underwater target detection is proposed to prevent missed detection of objects with different scales and overlapping occlusion.Firstly,deformable convolutions are introduced into the backbone network(deformable convolution network,DCN)to improve the feature extraction capability of the model by means of the adaptive deformation mechanism of convolution kernels.Secondly,a module combining atrous convolution and spatial pyramid,termed ASPF,is designed to expand the receptive field of the output feature map and improve the perception ability of the model for detecting underwater targets of multiple scales.Finally,the loss function is improved to optimize the training process of the model and improve detection accuracy.The improved algorithm is tested on the URPC data set,and the results show that its detection accuracy reaches 87.3%,which is 3.4%higher than that of the original YOLOv8 algorithm.Moreover,it can accurately detect underwater targets with different scales and overlapping occlusion.

关 键 词:水下目标检测 YOLOv8 深度学习 可变形卷积 WIoU 

分 类 号:TP391.41[自动化与计算机技术—计算机应用技术] TP183[自动化与计算机技术—计算机科学与技术]

 

参考文献:

正在载入数据...

 

二级参考文献:

正在载入数据...

 

耦合文献:

正在载入数据...

 

引证文献:

正在载入数据...

 

二级引证文献:

正在载入数据...

 

同被引文献:

正在载入数据...

 

相关期刊文献:

正在载入数据...

相关的主题
相关的作者对象
相关的机构对象