机械制造与自动化2024,Vol.53Issue(5) :229-233.DOI:10.19344/j.cnki.issn1671-5276.2024.05.048

基于改进卷积网络的铸件视觉检测

Casting Visual Detection Based on improved Convolutional Network

颜梦玫 杨冬平
机械制造与自动化2024,Vol.53Issue(5) :229-233.DOI:10.19344/j.cnki.issn1671-5276.2024.05.048

基于改进卷积网络的铸件视觉检测

Casting Visual Detection Based on improved Convolutional Network

颜梦玫 1杨冬平2
扫码查看

作者信息

  • 1. 福州大学 先进制造学院,福建 晋江 362251;中国科学院 海西研究院泉州装备制造研究中心,福建 晋江 362216
  • 2. 之江实验室 人工智能研究院混合增强智能研究中心,浙江 杭州 311101;中国科学院 海西研究院泉州装备制造研究中心,福建 晋江 362216
  • 折叠

摘要

为实现对铸件各种不同缺陷的自动诊断,设计一种视觉检测系统对铸造产品进行外观检测.利用多头自注意力模块对图像识别网络的下采样过程进行增强,使得经典图像分类网络能够获得全局特征信息,以提升模型对铸件体外观上细小裂纹和大范围毛刺的识别能力.实验结果表明:改进得到的卷积网络对铸件缺陷具有较高的识别准确率,能够较准确地判断铸件缺陷类型.

Abstract

For automatical diagnosing different defects in castings,a visual inspection system is designed to inspect the appearance of casting products.The multi-head self-attention module is used to enhance the downsampling process of the image recognition network,enabling the classic image classification network to obtain global feature information and promote the model's ability to identify small cracks and large-scale burrs on the appearance of castings.The experimental results show that the improved convolutional network has a higher recognition accuracy for casting defects,and can more accurately determine the type of casting defects.

关键词

缺陷检测/图像识别/卷积神经网络/深度学习/自注意力机制

Key words

defect detection/image recognition/convolutional neural network/deep learning/self-attention mechanism

引用本文复制引用

基金项目

国家自然科学基金项目(12175242)

出版年

2024
机械制造与自动化
南京机械工程学会 南京机电产业(集团)有限公司

机械制造与自动化

CSTPCD
影响因子:0.29
ISSN:1671-5276
参考文献量2
段落导航相关论文