高级检索

基于YOLO系列的小目标检测模型研究

Research on small target detection models based on YOLO

  • 摘要: 尽管目标检测技术已经日趋成熟,但小目标检测仍是研究中的难点。为解决模型难以检测小目标这一问题,提出了一种改进YOLO系列的目标检测模型。该模型引入了一种新的预分类策略,以减少各特征层之间的干扰,并在模型中嵌入Coord Attention模块,以增强特征提取能力,同时加入全局残差结构保留原始图像特征。在公开的VOC2007+2012联合数据集上的实验表明:① 改进后的YOLOv5和YOLOv7网络模型相较于原网络模型的平均精度分别提高了4.5%、0.8%;② 小目标检测精度分别提高了3.4%、6.3%,检测效果优于原网络模型。结果表明:提出的模型在目标检测任务中表现出了良好的鲁棒性和准确性。

     

    Abstract: Target detection technology has achieved significant progress, yet the detection of small target remains a challenging research area. To address the difficulty of detecting small target by models, an improved model for small target detection based on YOLO series was proposed. A pre-classification strategy was introduced to reduce interference between feature layers, and a Coord Attention module was embedded in the network to improve the performance of feature extraction. Additionally, a global residual structure was incorporated to ensure the integrity of the original image features. The model was tested on the publicly available VOC2007+2012 dataset, and the findings revealed that: ① the overall test set images of the improved YOLOv5 and YOLOv7 network models achieved average precision improvements of 4.5% and 0.8%, respectively, compared to the original network models; ② the detection accuracy for small targets was increased by 3.4% and 6.3%, respectively, outperforming the original network models. The results indicate that the proposed model exhibits robustness and accuracy in target detection tasks.

     

/

返回文章
返回