基于YOLOv7-TSCR的连铸坯表面缺陷检测
DOI:
作者:
作者单位:

1.华北理工大学电气工程学院;2.华北理工大学冶金与能源学院;3.河北科技大学材料科学与工程学院

作者简介:

通讯作者:

中图分类号:

TP391.4

基金项目:

51904107


Author:
Affiliation:

Fund Project:

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

    为解决连铸生产过程中铸坯表面缺陷检测准确率低、检测速度慢、模型参数量大难以部署等问题,提出一种融合重参数化和注意力机制的轻量化铸坯表面缺陷检测算法YOLOv7-TSCR。首先,利用Mish和SiLU激活函数、SimAM注意力机制,构建了改进的高效层聚合模块ELAN-S,有效增强了对多尺度缺陷特征的提取;其次,设计了C2f_RG模块改进特征融合网络,减少参数量的同时获得更丰富的梯度流信息,增强特征融合能力;最后,根据采集实际生产中的缺陷图像,构建铸坯缺陷数据集并进行验证。实验结果表明:YOLOv7-TSCR相较其它网络模型检测效果显著提升,在模型参数量减小的情况下,精确率达93.5%,平均精度均值提高了2.8%,检测速度可达120FPS。同时,在NEU-DET公开数据集上进行泛化性对比实验,证明了算法具有较强的泛化性。改进算法在保证较高检测精度的基础上,具有较快的检测速度和较小的参数量,为铸坯表面缺陷的高效检测提供了技术参考。

    Abstract:

    参考文献
    相似文献
    引证文献
引用本文
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2024-03-04
  • 最后修改日期:2024-04-11
  • 录用日期:2024-04-24
  • 在线发布日期:
  • 出版日期:
文章二维码