现代制造工程Issue(5)：113-120,94,9.DOI:10.16731/j.cnki.1671-3133.2024.05.015
基于MFA-UNet的铜制螺纹零件外表面缺陷检测
Copper threaded part surface defect detection algorithm based on MFA-UNet
摘要
Abstract
In industrial settings,detecting surface defects on copper threaded parts often faces challenges of low efficiency and poor accuracy.To address this,it proposes a copper threaded part surface defect detection algorithm based on MFA-UNet(Multi-Scale Features and Attention Fused UNet).Firstly,a dual down sampling module is designed,utilizing both ordinary convolution and dilated convolution to enhance the model's feature extraction capabilities.Secondly,a compound spatial attention module is integrated into the skip-connection part to improve the model's ability to extract spatial and edge information.Subsequently,a squeeze and excitation module is incorporated during the upsampling process to enhance the model's expressive power and feature selection ability.Lastly,it proposes a similarity comparison algorithm that measures the similarity between segmented images and mask images to obtain the defect detection results.Experimental results demonstrate that the proposed segmentation model a-chieves a PA of 94.81％and an MIoU of 93.78％on the copper threaded part defect detection dataset.The defect detection ac-curacy of the proposed algorithm reaches 98.9％,meeting the requirements for industrial field applications.关键词
零件缺陷检测/图像分割/注意力机制/相似度对比Key words
part defect detection/image segmentation/attention mechanism/similarity comparison分类
信息技术与安全科学引用本文复制引用
马涛,李敬兆..基于MFA-UNet的铜制螺纹零件外表面缺陷检测[J].现代制造工程,2024,(5):113-120,94,9.基金项目
国家自然科学基金资助项目(51874010) （51874010）
