计量学报

北大核心,JST,CSCD扩展版,WJCI,

国内刊号:11-1864/TB

国际刊号:1000-1158

计量学报杂志2022年第11期:基于改进YOLOv4算法的轮毂表面缺陷检测

发布日期:

作者:吴凤和,崔健新,张宁,张志良,张会龙,郭保苏

单位:燕山大学机械工程学院,河北 秦皇岛 066004

关键词:计量学;轮毂;缺陷检测;改进YOLOv4算法;细化U型网络

基金:国家重点研发计划(2020YFB1700803);河北省高等学校科学技术研究重点项目(ZD2020156)

汽车轮毂加工过程中产生的表面缺陷严重影响整车的美观性及服役性能,针对人工检测效率低、漏检率高的问题,提出一种基于改进YOLOv4算法的轮毂表面缺陷检测方法。构建了轮毂缺陷数据集,其包含6种表面缺陷,由2346张4928×3264pixel的图像组成;采用K-means方法进行先验框聚类,并针对YOLOv4算法在纤维、粘铝等小尺度缺陷上检测精度不足问题,在原网络Neck部分引入细化U型网络模块(TUM)和注意力机制,用于增强有效特征并抑制无效特征,强化多尺度特征提取与融合,改善特征处理过程中可能存在的小目标信息丢失问题;基于该数据集,训练并测试不同算法的缺陷检测性能并验证改进模块的有效性。结果表明,该方法大幅提升了粘铝等小尺寸缺陷的检测能力,缺陷检测平均精度达到85.8%,与多种算法相比较检测精度最高。

来源:2022年第11期

《计量学报》期刊编辑部

查看计量学报杂志2022年第11期

联系我们

  • 地址:北京市朝阳区北三环东路18号
  • 电话:010-64271480

咨询工作人员