注意力机制优化 RetinaNet 的密集工件检测方法研究
DOI:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

TP391. 41

基金项目:

西安市科技局高校人才服务企业项目(GXYD75)、陕西省科技厅工业领域一般项目(2018GY173)资助


Research on dense workpiece detection method based on attentional mechanism optimization RetinaNet
Author:
Affiliation:

Fund Project:

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

    针对密集工件存在相似度高、排列无序的特性导致检测难度大的问题,提出了一种注意力机制优化 RetinaNet 的密集工 件检测方法。 首先将注意力机制引入到 RetinaNet 主干特征提取网络以减少干扰物对检测效果的影响,提高神经网络的特征提 取能力;然后利用 Soft-NMS 构建新的预测框提高重叠定位精度;最后通过迁移学习的方法训练数据集,提高模型训练效率。 在 密集工件数据集上验证该方法的有效性,实验结果表明,改进后的方法检测精度达到了 98. 11%,相较于改进前提高了 2. 59%, 单张图片检测速度达到了 0. 026 s,该方法能够满足实际工业生产过程中精准检测工件的目的,在保证速度的同时降低了漏检 率和误检率。

    Abstract:

    In order to solve the problem of difficult detection due to the existence of dense workpiece with high similarity and disorderly arrangement, an attention mechanism is proposed to optimize RetinaNet’ s dense workpiece detection method. Firstly, the attention mechanism is introduced into the RetinaNet backbone feature extraction network to reduce the influence of interfering objects on the detection effect and improve the feature extraction ability of the neural network. then a new predictive box is constructed using Soft-NMS to improve the overlap localization accuracy. Finally, the dataset is trained by transfer learning method to improve the model training efficiency. The method effectiveness is verified on the produced dense workpiece dataset; Experimental results show that the detection accuracy of the improved method reaches 98. 11%, which is 2. 59% higher in comparison with that before the improvement. The detection speed of a single picture is up to 0. 026s. The proposed method can meet the purpose of accurate detection of workpiece in actual industrial production process, which can reduce the rate of missed and false detection and assure the speed simultaneously.

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

徐 健,陆 珍,刘秀平,张立昌,闫焕营.注意力机制优化 RetinaNet 的密集工件检测方法研究[J].电子测量与仪器学报,2022,36(1):237-235

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