基于目标识别的钢材缺陷检测方法

作     者：程国建 卞晨亮 CHENG Guo-jian;BIAN Chen-liang

作者机构：西安石油大学计算机学院陕西西安710065 

基　　金：国家自然科学基金青年科学基金项目(41301480) 国家自然科学基金项目(62002286) 

出 版 物：《计算机工程与设计》 (Computer Engineering and Design)

年 卷 期：2022年第43卷第10期

页      码：2977-2983页

摘      要：为实现热轧钢板表面缺陷检测,设计一个端到端的缺陷检测算法YOLOv5-MD。为达到各种样本数量均衡分布,使用数据增强算法对钢材缺陷数据集处理,使用通道混洗模块缓解特征融合中计算量的占用问题;引入注意力机制和加权边界框融合(weighted-boxes-fusion,WBF)消除重叠度较高的预测框;在PANet(path aggregation network)模块中添加残差的跳跃结构并提取多尺度特征来获取更多的细节图像信息。经实验结果验证,所提模型的平均检测精度(mean average precision,mAP)可达97.2%,相较于YOLOv5提高了10%左右,检测速度和YOLOv5持平,是Faster RCNN的7.1倍,是SSD的1.7倍,验证了该方法的可行性、有效性。

主 题 词：缺陷检测 聚类 数据精准增强 通道转换卷积 注意力机制 加权边界框融合 路径聚合网络 

学科分类：08[工学] 080203[080203] 0802[工学-机械学] 

D　O　I：10.16208/j.issn1000-7024.2022.10.037

馆 藏 号：203114633...