改进YOLOv5的轻量级带钢表面缺陷检测
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:


Lightweight Strip Steel Defect Detection Based on Improved YOLOv5
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 增强出版
  • |
  • 文章评论
    摘要:

    针对工业场景下设备资源有限的情况, 提出一种改进YOLOv5的轻量化带钢缺陷检测模型. 首先, 使用ShuffleNetv2代替主干特征提取网络, 优化模型参数量和运行速度; 其次, 采用轻量级上采样算子CARAFE (content-aware reassembly of features), 在增大感受野的同时进一步降低参数和计算量; 同时引入GSConv层, 在保证语义信息的同时平衡模型准确性与检测速度; 最后, 设计一种跨层级特征融合机制, 提高网络的检测精度. 实验结果表明, 改进后的模型的平均检测精度为78.5%, 相较于原始YOLOv5算法提升了1.4%; 模型计算量为10.9 GFLOPs, 参数量为5.88×106, 计算量和参数量分别降低31%和15.4%; 检测速度为49 f/s, 提升了3.5 f/s. 因此, 改进后的模型提高了检测精度和检测速度, 并且大幅降低了模型的计算量和参数量, 能够满足对带钢表面缺陷进行实时检测.

    Abstract:

    For limited equipment resources in industrial scenarios, a lightweight strip steel defect detection model based on improved YOLOv5 is proposed. First, ShuffleNetv2 is used to replace the backbone feature extraction network to optimize model parameter amount and running speed; secondly, the lightweight up-sampling operator, namely content-aware reassembly of features (CARAFE) is used to further reduce parameters and calculation amount while increasing the receptive field. At the same time, the GSConv layer is introduced to balance the model accuracy and detection speed while ensuring semantic information. Finally, a cross-level feature fusion mechanism is designed to improve the detection accuracy of the network. The experimental results show that the mean average precision of the improved model is 78.5%, which is 1.4% higher than the original YOLOv5 algorithm. The calculation amount of the model is 10.9 GFLOPs; the parameter amount is 5.88×106; the calculation and parameter amounts are reduced by 31% and 15.4%, respectively; the detection speed is 49 f/s, which is increased by 3.5 f/s. Therefore, the improved model improves the detection accuracy and speed and greatly reduces the calculation and parameter amounts of the model, which can ensure the real-time detection of surface defects of strip steel.

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

张政超.改进YOLOv5的轻量级带钢表面缺陷检测.计算机系统应用,2023,32(6):278-285

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2022-12-05
  • 最后修改日期:2023-01-06
  • 录用日期:
  • 在线发布日期: 2023-04-20
  • 出版日期:
您是第位访问者
版权所有:中国科学院软件研究所 京ICP备05046678号-3
地址:北京海淀区中关村南四街4号 中科院软件园区 7号楼305房间,邮政编码:100190
电话:010-62661041 传真: Email:csa (a) iscas.ac.cn
技术支持:北京勤云科技发展有限公司

京公网安备 11040202500063号