高级检索

    基于深度学习的小目标检测识别方法

    A Small Object Detection and Recognition Method Based on Deep Learning

    • 摘要: 针对复杂背景下现有基于网络摄像头和旋翼无人机采集的可见光视频中的小目标检测精度低的问题,提出了一种基于改进YOLOv5的小目标检测识别方法。通过将SPD-Conv模块、协调注意力(Coordinate Attention,CA)模块、转置卷积、优化锚框、改进的Alpha-loU损失函数等策略引入YOLOv5架构,可有效提高对小目标的检测能力。实验结果表明,所提改进YOLOv5具备小目标检测能力,精度、召回率和mAP指标分别可达到82.14%、75.96和82.41%,推理速度小于25 ms,参数量小于40 M。实验结果验证了所提方法的有效性和实用性,该方法具有广阔的应用前景。

       

      Abstract: In view of the problem of low detection accuracy of small targets in visible-light videos collected by network cameras and rotary-wing drones in complex backgrounds, a small target detection and recognition method based on improved YOLOv5 is proposed. By introducing strategies such as the SPD-Conv module, coordinate attention (CA) module, transposed convolution, optimized anchor boxes, and an improved Alpha - IoU loss function into the YOLOv5 architecture, the detection ability for small targets can be effectively enhanced. Experimental results demonstrate that the proposed improved YOLOv5 has the ability to detect small targets. The precision, recall rate, and mAP indicators can reach 82.14%, 75.96, and 82.41% respectively. The inference speed is less than 25 ms, and the number of parameters is less than 40 M. The experimental results verify the effectiveness and practicality of the proposed method, which has broad application prospects.

       

    /

    返回文章
    返回