基于SlimYOLO的控制箱零件检测方法
DOI:
CSTR:
作者:
作者单位:

武汉工程大学 智能机器人湖北省重点实验室 武汉 430205

作者简介:

通讯作者:

中图分类号:

TP391.4

基金项目:

国家自然科学基金资助项目(No.62171328);国家自然科学基金资助项目(No.62102292)


Detection method of electrical cabinet parts based on SlimYOLO
Author:
Affiliation:

Hubei Provincial Key Laboratory of Intelligent Robot, Wuhan Institute of Technology, Wuhan 430000, China

Fund Project:

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

    控制箱零件检测是控制箱生产过程中的重要环节。采用机器视觉方法可自动识别控制箱内零件的类别及安装位置,及时检测控制箱装配缺陷。然而现有目标检测深度学习模型时效性较低,难以满足控制箱零件在线实时检测需求。本文对YOLOv4目标检测模型进行剪枝和优化,提出了轻量级的目标检测模型SlimYOLO。SlimYOLO改进了特征提取网络结构,压缩了冗余特征层,提高了模型推理速度。同时采用Kmeans++聚类算法对模型anchor框参数进行聚类分析,提升了模型对控制箱的检测效果。基于自主构建的控制箱零件数据集开展了多项对比实验研究,SlimYOLO的平均检测精度为98.08%,较YOLOv4提升0.58%,模型体积缩小9.8%,参数量减少了700万,推理速度提升了10%,为实际工业场景中控制箱零件的快速智能化检测奠定了基础。

    Abstract:

    The detection of electrical cabinet parts is an important part of the production of electrical cabinets. Machine vision is used to automatically identify the type and installation location of parts in the electrical cabinet, and to detect the assembly defects of the electrical cabinet in time. However, the existing object detection in-depth learning model has low timeliness, which makes it difficult to meet the online detection requirements of electrical cabinet parts. In this paper, the YOLOv4 object detection model is pruned and optimized, and a lightweight object detection model SlimYOLO is proposed. SlimYOLO improves the feature extraction network structure, compresses the redundant feature layer, and improves the detection speed of the model. At the same time, the Kmeans++ clustering algorithm is used to cluster anchor box parameters, which improves the detection effect of the model for electrical cabinet parts. Based on the self-built data set of electrical cabinet parts, an experimental study was carried out. The average detection accuracy of SlimYOLO is 98.08%, which is 0.58% higher than YOLOv4, the model volume is reduced by 9.8%, the parameter is reduced by about 7 million, and the detection speed is increased by 10%, which lays a foundation for the fast and intelligent detection of electrical cabinet parts in the actual industrial scene.

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

冯晨光,魏巍,陈 灯,张彦铎,刘玮,栗娟.基于SlimYOLO的控制箱零件检测方法[J].电子测量技术,2022,45(17):120-126

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