现代制造工程 ›› 2023, Vol. 509 ›› Issue (2): 1-9.doi: 10.16731/j.cnki.1671-3133.2023.02.001

• 试验研究 •    

基于SimAM模块与ResNet34网络的混合缺陷检测模型

朱传军;刘荣光;成佳闻;梁泽启;王林琳   

  • 发布日期:2023-03-20
  • 基金资助:
    国家自然科学基金资助项目(51965031); 甘肃省教育厅产业支撑计划项目(2022CYZC-68); 甘肃省自然科学基金项目(20JR5RA376); 甘肃省教育科技创新项目(2022A-160)

  • Published:2023-03-20

摘要: 针对现表面缺陷检测方法准确率低、需要进行复杂的特征设计、特征泛化性不强、参数多和识别速度慢等问题,在残差网络卷积模块之后采用自适应全局平均池化,有效降低了分类器的特征维度,减少了信息冗余。将无参注意力机制模块SimAM与ResNet34网络相结合用于缺陷检测,并对不同组合结构进行研究,提出ResNet34_s_e和ResNet34_m这2种混合网络模型,该2种混合网络模型均不增加原始网络参数量。在东北大学钢铁缺陷标准数据集上进行实验,对数据集使用镜像、翻转等数据增广策略,防止模型过拟合。通过对比发现,ResNet34_s_e混合网络模型能够有效加快训练过程中误差的下降趋势,提升分类准确率。最后在武汉某制造车间采集的冲压件缺陷数据集上验证该混合网络模型的泛化性能。测试集正确率由88.34%提高到了89.19%,有效提升了车间冲压件缺陷检测准确率。

关键词: 缺陷识别; 残差网络; 数据增广; SimAM模块

中图分类号: 


版权所有 © 《现代制造工程》编辑部 
地址:北京市东城区东四块玉南街28号 邮编:100061 电话:010-67126028 电子信箱:2645173083@qq.com
本系统由北京玛格泰克科技发展有限公司设计开发 技术支持:support@magtech.com.cn
访问总数:,当日访问:,当前在线: