高级检索

面向无人机航拍图像小目标检测方法

A Small Target Detection Method for Unmanned Aerial Vehicle Aerial Photography Images

  • 摘要: 针对航拍图像目标检测中小目标特征模糊问题,提出一种改进YOLO_v5x的目标检测算法。通过在YOLO_v5x的主干和颈部网络中添加空间到深度(space-to-depth,SPD)模块来减少细粒度信息丢失;在检测输出端添加1个小目标预测头,提高算法学习低分辨率特征的效率;引入协调注意力(coordinate attention,CA)机制,将横向和纵向的位置信息编码到通道注意中,增强网络对不同维度特征的提取能力;在完整交并比 (complete-intersection over union,CIOU)损失函数的基础上引入Alpha交并比(α−IOU)损失函数,获得更准确的边界框回归,实现图像中目标更精确的定位。通过在Visdrone数据集上对改进YOLO_v5x算法进行训练和对比实验,结果表明:相比于原YOLO_v5x,改进目标检测算法的平均检测精度提升了7.8%,小目标检测的平均精度达23.9%,能够有效识别无人机航拍图中的小目标;相比于RetinaNet 、YOLOX-S、Grid-RCNN等目标检测算法,改进目标检测算法的小目标检测平均精度最高,在当前主流检测小目标算法中达到先进水平。

     

    Abstract: Aiming at the problem of fuzzy features of small targets in aerial image detection, an improved YOLO_v5x target detection method was proposed. A space-to-depth (SPD) module was added to the backbone and neck network of YOLO_v5x to reduce the loss of fine-grained information, and a small target prediction head was added to the detection output to improve the efficiency of the algorithm in learning low-resolution features. At the same time, the coordinate attention (CA) mechanism was introduced to encode the horizontal and vertical position information into the channel attention to enhance the ability of the network to extract different dimensional features. In order to improve the target positioning accuracy, the Alpha intersection over union (α−IOU) loss function was introduced based on the complete-intersection over union (CIOU) loss function. To obtain more accurate bounding box regression, to achieve more accurate target positioning in the image. Through training and comparative experiments on the improved YOLO_v5x algorithm on the Visdrone datasets. The results show that compared with the original YOLO_v5x, the average detection accuracy of the improved target detection algorithm was increased by 7.8%, and the average detection accuracy of small target detection was up to 23.9%, which can effectively identify small targets in unmanned aerial vehicle aerial photos. Compared with other target detection algorithms such as RetinaNet and YOLOX-S, the average precision of small target detection was the highest in the improved target detection algorithm, reaching the advanced level among the current mainstream small target detection algorithms.

     

/

返回文章
返回