-
题名基于改进YOLOv4算法的轮毂表面缺陷检测
被引量：6
- 1
-
-
作者
吴凤和
崔健新
张宁
张志良
张会龙
郭保苏
-
机构
燕山大学机械工程学院
中信戴卡股份有限公司
-
出处
《计量学报》
CSCD
北大核心
2022年第11期1404-1411,共8页
-
基金
国家重点研发计划(2020YFB1700803)
河北省高等学校科学技术研究重点项目(ZD2020156)。
-
文摘
汽车轮毂加工过程中产生的表面缺陷严重影响整车的美观性及服役性能,针对人工检测效率低、漏检率高的问题,提出一种基于改进YOLOv4算法的轮毂表面缺陷检测方法。构建了轮毂缺陷数据集,其包含6种表面缺陷,由2346张4928×3264 pixel的图像组成;采用K-means方法进行先验框聚类,并针对YOLOv4算法在纤维、粘铝等小尺度缺陷上检测精度不足问题,在原网络Neck部分引入细化U型网络模块(TUM)和注意力机制,用于增强有效特征并抑制无效特征,强化多尺度特征提取与融合,改善特征处理过程中可能存在的小目标信息丢失问题;基于该数据集,训练并测试不同算法的缺陷检测性能并验证改进模块的有效性。结果表明,该方法大幅提升了粘铝等小尺寸缺陷的检测能力,缺陷检测平均精度达到85.8%,与多种算法相比较检测精度最高。
-
关键词
计量学
轮毂
缺陷检测
改进YOLOv4算法
细化u型网络
-
Keywords
metrology
wheel hub
defect detection
improved YOLOv4 algorithm
TuM
-
分类号
TP92
[自动化与计算机技术]
-
