多尺度特征在 YOLO 算法中的应用研究
DOI:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

TP391. 4;TN911. 73

基金项目:

芜湖市科技计划(2020cg12)、安徽工程大学校级科研项目(xjky02201909)资助


Application research of multi-scale features in YOLO slgorithm
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对 YOLO 算法在下采样过程中丢失了部分大尺寸特征图的有效信息,从而导致在检测任务中因目标定位不够精准而 影响模型整体检测精度的问题。 提出利用多尺度特征融合的方法来解决 YOLO 定位不精准的问题,首先,对 YOLO 算法的网络 模型进行修改,利用 YOLO 网络模型中不同尺寸特征图具有不同特征属性的特点,融合不同尺寸特征图来提高检测网络对目标 的定位精度;其次在预训练模型的基础上对修改后的网络模型进行重新训练;最后在计算机中对训练好的模型进行检测试验。 实验结果表明,基于多尺度特征的 YOLO 目标检测算法在精确率上相对于 YOLO 目标检测算法提高了 3. 02%,mAP 提高 了 1. 53%。

    Abstract:

    YOLO algorithm loses part of the effective information of large-size feature maps in the downsampling process, which leads to the problem of insufficient target location in the detection task, which affects the overall detection accuracy of the model. This paper proposes the use of multi-scale feature fusion to solve the problem of inaccurate location of YOLO; First, modify the network model of YOLO algorithm, use different size feature maps in the YOLO network model with different feature attributes, and merge different size feature maps to improve the location accuracy of the detection network to the target; second, based on the pre-training model Re-train the modified network model on the last; finally, test the trained model in the computer. Experimental results show that the YOLO target detection algorithm based on multi-scale features improves the Accuracy rate by 3. 02% and improves mAP by 1. 53% compared with YOLO target detection algorithm.

    参考文献
    相似文献
    引证文献
引用本文

徐晓光,李 海.多尺度特征在 YOLO 算法中的应用研究[J].电子测量与仪器学报,2021,35(6):96-101

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2023-02-27
  • 出版日期: