DEFECT RECOGNITION OF METAL COMPONENTS BASED ON TRANSFER LEARNING AND FEATURE FUSION

被引:0
作者
Li Yuanyuan [1 ]
Zhao Junren [1 ]
Liu Hailong [1 ]
Chen Xi [2 ]
机构
[1] Univ Elect Sci & Technol China, Sch Resources & Environm, Chengdu 611731, Peoples R China
[2] Chinese Acad Sci, Xinjiang Inst Ecol & Geog, Urumqi 830011, Peoples R China
来源
2022 19TH INTERNATIONAL COMPUTER CONFERENCE ON WAVELET ACTIVE MEDIA TECHNOLOGY AND INFORMATION PROCESSING (ICCWAMTIP) | 2022年
关键词
Defect recognition; Image classification; Neural network; Transfer learning;
D O I
10.1109/ICCWAMTIP56608.2022.10016502
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
While Artificial Intelligence continues to permeate industrial production, the recognition of surface defects in metal components is constrained by the difficulty of identifying subtle and easily confused surface defects and the lack of a sufficient number of samples for training the recognition model. This study combines five typical neural networks-AlexNet, GoogLeNet, ResNet, Xception, and ResNeXt - by using transfer learning based on the ImageNet dataset, and applies multi-scale convolution, shortcut connection, and mixed pooling to enhance recognition performance. This paper is an algorithm for defect recognition based on transfer learning, feature fusion, and enhanced generalization. The proposed method can accurately recognize defects on the surface of metal components, with a higher accuracy of 97.81% and a true positive rate of 98.32%. Its accuracy on the test dataset varied from 87.69% to 97.81%, which indicates that our model is significantly better than traditional machine learning methods and the original neural network. It can also provide a reference for intelligent defect recognition in metal components.
引用
收藏
页数:8
相关论文
共 31 条
[1]  
Ballester P, 2016, AAAI CONF ARTIF INTE, P1124
[2]   Fabric defect detection by Fourier analysis [J].
Chan, CH ;
Pang, GKH .
IEEE TRANSACTIONS ON INDUSTRY APPLICATIONS, 2000, 36 (05) :1267-1276
[3]   Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation [J].
Chen, Liang-Chieh ;
Zhu, Yukun ;
Papandreou, George ;
Schroff, Florian ;
Adam, Hartwig .
COMPUTER VISION - ECCV 2018, PT VII, 2018, 11211 :833-851
[4]  
Chen Q., 2020, NEURAL COMPUTING ADV, P420, DOI [10.1007/978-981-15-7670-6_35, DOI 10.1007/978-981-15-7670-6_35]
[5]   Xception: Deep Learning with Depthwise Separable Convolutions [J].
Chollet, Francois .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :1800-1807
[6]  
Ghifary M, 2014, LECT NOTES ARTIF INT, V8862, P898, DOI 10.1007/978-3-319-13560-1_76
[7]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[8]  
[胡正平 Hu Zhengping], 2020, [信号处理, Journal of Signal Processing], V36, P58
[9]   Dragonflies segmentation with U-Net based on cascaded ResNeXt cells [J].
Hurtik, Petr ;
Ozana, Stanislav .
NEURAL COMPUTING & APPLICATIONS, 2021, 33 (09) :4567-4578
[10]  
Jing Junfeng, 2018, PROCEEDING 10 INT C, P68