计算机工程与应用 ›› 2022, Vol. 58 ›› Issue (10): 208-215.DOI: 10.3778/j.issn.1002-8331.2010-0251

• 图形图像处理 • 上一篇    下一篇

融合DBSCAN的改进YOLOv3目标检测算法

李云红,张轩,李传真,苏雪平,聂梦瑄,毕远东,谢蓉蓉   

  1. 西安工程大学 电子信息学院,西安 710048
  • 出版日期:2022-05-15 发布日期:2022-05-15

Improved YOLOv3 Target Detection Algorithm Combined with DBSCAN

LI Yunhong, ZHANG Xuan, LI Chuanzhen, SU Xueping, NIE Mengxuan, BI Yuandong, XIE Rongrong   

  1. School of Electronic Information, Xi’an Polytechnic University, Xi’an 710048, China
  • Online:2022-05-15 Published:2022-05-15

摘要: 针对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聚类算法

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.

Key words: YOLOv3, convolutional neural network, target detection, DBSCAN clustering algorithm