首页 | 官方网站   微博 | 高级检索  
     

融合DBSCAN的改进YOLOv3目标检测算法
引用本文:李云红,张轩,李传真,苏雪平,聂梦瑄,毕远东,谢蓉蓉. 融合DBSCAN的改进YOLOv3目标检测算法[J]. 计算机工程与应用, 2022, 58(10): 208-215. DOI: 10.3778/j.issn.1002-8331.2010-0251
作者姓名:李云红  张轩  李传真  苏雪平  聂梦瑄  毕远东  谢蓉蓉
作者单位:西安工程大学 电子信息学院,西安 710048
基金项目:陕西省教育厅自然科学基础研究计划项目;国家级大学生创新创业训练计划项目;国家自然科学基金;陕西省科技厅自然科学基础研究重点项目
摘    要:针对YOLOv3(you only look once)检测算法对小目标、遮挡目标检测时存在识别率低和识别精度不高的问题,提出一种融合DBSCAN(density-based spatial clustering of applications with noise)的改进YOLOv3目标检测算法。首先在YOLOv3网络中增加DBSCAN聚类算法,其次对检测目标进行提取,实现数据集多尺度聚类,得到初代特征图,然后通过改进[K]-means聚类算法确定锚点位置达到更好的聚类,最后在VOC2007+2012数据集和MS-COCO数据集上对改进YOLOv3算法进行训练和测试。实验结果表明改进的YOLOv3算法使检测目标在VOC数据集和MS-COCO数据集上mAP(mean average precision)分别提高了14.9个百分点和12.5个百分点。与其他深度学习目标检测算法相比,改进YOLOv3检测算法具有更好的检测效果,同时具有良好移植性和更好的鲁棒性。

关 键 词:YOLOv3  卷积神经网络  目标检测  DBSCAN聚类算法  

Improved YOLOv3 Target Detection Algorithm Combined with DBSCAN
LI Yunhong,ZHANG Xuan,LI Chuanzhen,SU Xueping,NIE Mengxuan,BI Yuandong,XIE Rongrong. Improved YOLOv3 Target Detection Algorithm Combined with DBSCAN[J]. Computer Engineering and Applications, 2022, 58(10): 208-215. DOI: 10.3778/j.issn.1002-8331.2010-0251
Authors:LI Yunhong  ZHANG Xuan  LI Chuanzhen  SU Xueping  NIE Mengxuan  BI Yuandong  XIE Rongrong
Affiliation:School of Electronic Information, Xi’an Polytechnic University, Xi’an 710048, China
Abstract:Aiming at the problems of low recognition rate and low recognition accuracy when the YOLOv3(you only look once) detection algorithm detects small targets and occluded targets, an improved YOLOv3 algorithm is proposed in combination with DBSCAN(density-based spatial clustering of applications with noise) for target detection. Firstly, DBSCAN clustering algorithm is added to YOLOv3 network, and the detection target is extracted to achieve multi-scale clustering of the dataset to obtain the first generation feature map, and then the anchor point location is determined by improving [K]-means clustering algorithm to achieve better clustering. Finally, the improved YOLOv3 algorithm is trained and tested on VOC2007+2012 dataset and MS-COCO dataset. The experimental results show that the improved YOLOv3 algorithm increases the mAP of detection target by 14.9 percentage points and 12.5 percentage points on the VOC dataset and MS-COCO dataset, respectively. The improved YOLOv3 detection algorithm has better detection results in comparison with other deep learning target detection algorithms, as well as good portability and better robustness.
Keywords:YOLOv3   convolutional neural network   target detection   DBSCAN clustering algorithm  
本文献已被 万方数据 等数据库收录!
点击此处可从《计算机工程与应用》浏览原始摘要信息
点击此处可从《计算机工程与应用》下载免费的PDF全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司    京ICP备09084417号-23

京公网安备 11010802026262号