基于改进YOLOv5的高空螺母识别算法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

TP391.4

基金项目:

国家自然科学基金项目(面上项目,重点项目,重大项目)


Research on high-altitude nut recognition algorithm based on improved YOLOv5
Author:
Affiliation:

Fund Project:

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

    为了提升高空螺母的识别准确率,减少螺栓螺母的误检、漏检率,本研究提出了一种基于改进YOLO V5的高空螺母识别模型。首先,在骨干网络端添加了新型注意力机制EMA,以此融合更多的信息。其次,将颈部网络的PANet更换为BiFPN,以加强网络的特征提取能力。最后,将原损失函数CIOU更换为SIOU,以加快模型的收敛速度并提高模型的分类准确率。结果表明,相比于YOLO V5原模型,改进后的模型拥有更好的性能,其中准确率提升了0.92%,召回率提升了0.16%,平均精度1(mAP_0.5)提升了0.53%,平均精度2(mAP_0.5:0.95)提升了2.26%。再用改进前后的模型进行实际识别对比实验,结果表明,改进后的模型识别效果更好,漏检、误检率下降,实际的识别率更高。改进后的模型能够很好地满足高空螺母的识别和图像数据采集,也为后续的螺母维护提供了数据基础。

    Abstract:

    In order to improve the recognition accuracy of high-altitude nuts and reduce the false detection rate of bolts and nuts, a high-altitude nut recognition model based on improved YOLO V5 is proposed in this study. Firstly, a new attention mechanism EMA was added to the backbone network to integrate more information. Secondly, in order to enhance the network"s feature extraction capability, BiFPN was used to replace the PANet of the neck network. Finally, SIOU was used to replace the original loss function CIOU to accelerate the convergence of the model and improve its classification accuracy. The results show that the improved model has better performance than the original YOLO V5 model. The accuracy of the improved model increased by 0.92%. The recall increased by 0.16%. The average precision 1 (mAP:0.5) increased by 0.53%. And the average precision 2 (mAP:0.95) increased by 2.26%. An actual recognition comparison experiment between the improved model and the original YOLO V5 model is being carried out. The experimental results that the improved model has better recognition performance, which reduces the missed detection rate and the false detection rate, and improves the actual recognition rate. The improved model can well meet the recognition and image data acquisition of high-altitude nuts. And it also provide a data foundation for subsequent nut maintenance.

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

孟芳芳,田孝壮,方薇,等. 基于改进YOLOv5的高空螺母识别算法[J]. 科学技术与工程, 2025, 25(1): 262-269.
Meng Fangfang, Tian Xiaozhuang, Fang Wei, et al. Research on high-altitude nut recognition algorithm based on improved YOLOv5[J]. Science Technology and Engineering,2025,25(1):262-269.

复制
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2024-04-07
  • 最后修改日期:2024-04-27
  • 录用日期:2024-05-02
  • 在线发布日期: 2025-01-13
  • 出版日期:
×
亟待确认版面费归属稿件,敬请作者关注