Journal of Jilin University(Engineering and Technology Edition) ›› 2023, Vol. 53 ›› Issue (5): 1449-1457.doi: 10.13229/j.cnki.jdxbgxb.20210855

Previous Articles    

Surface defect detection algorithm of magnetic tiles based on multi⁃branch convolutional neural network

Pei-yong LIU1,2(),Jie DONG1,Luo-feng XIE1,Yang-yang ZHU1,Guo-fu YIN1()   

  1. 1.School of Mechanical Engineering,Sichuan University,Chengdu 610065,China
    2.Chengdu Aeronautic Polytechnic,Chengdu 610100,China
  • Received:2021-08-13 Online:2023-05-01 Published:2023-05-25
  • Contact: Guo-fu YIN E-mail:peiyongliu@126.com;gfyin@scu.edu.cn

Abstract:

Aiming at the problem of great difficulty and low accuracy in the detection of magnetic tile surface defects(DMTSD), a novel algorithm of DMTSD is proposed. In this paper, a multi-branch network structure is designed, and a convolutional neural network which can effectively extract the features of the magnetic tile images is constructed in each branch, and then the attention module is introduced to highlight the important features of the image. Finally, the correlation of the intra-class features and the difference of the inter-class features is maximized through discriminant correlation analysis, and the optimized fusion feature of magnetic tile images is obtained by concatenation fusion. The performance of the algorithm is tested on the magnetic tile image data set, and the test accuracy is up to 99.90%. In the actual detection work, the detection accuracy of the algorithm remains above 99%, and the detection speed reaches 129 pcs/min. The results of test and operation show that the algorithm has the advantage of high detection accuracy and stable performance, and can meet the requirements of real-time online detection in mass production of magnetic tiles.

Key words: computer application, magnetic tile, defect detection, multi-branch, convolutional neural network, feature fusion

CLC Number: 

  • TP391

Fig.1

Overall structure of the MBIVGG network"

Fig.2

Three dimensional models of typical magnetic tiles"

Fig.3

Schematic diagram of image acquisition for larger radian magnetic tiles"

Fig.4

Performance comparison of four classical CNN for surface defect detection of magnetic tiles"

Table 1

Performance test of four classical CNN in surface defect detection of magnetic tiles"

网络名称训练精度测试精度平均测试 时间t1/s单张图像平均测试时间t2/s最大测试 时间t3/s最大平均 测试时间t4/s检测速度n/(块·min-1
VGG160.99120.9922.930.02933.450.0345217
AlexNet0.98420.9841.990.01992.580.0258290
ResNet180.98920.9892.340.02342.390.0239313
GoogLeNet0.98480.9854.690.04694.780.0478156

Fig.5

Structure of IVGG convolutional neural network"

Fig.6

Training results of IVGG network"

Fig.7

Attention mechanism of CBAM"

Fig.8

Classification of magnetic tiles"

Fig.9

Flexible intelligent detection equipment for magnetic tile surface defects"

Table 2

Sample partition"

数据集线状缺陷块状缺陷正常图像合计
训练集9609609602880
验证集320320320960
测试集320320320960

Fig.10

Training results of MBIVGG network"

Table 3

Comparison of test results(VGG16/MBIVGG)"

实际类别检测结果样品数量
合格线状缺陷块状缺陷
合格289/2997/04/1300
线状缺陷16/3279/2965/1300
块状缺陷9/17/0284/299300
正确拒绝率3.67% / 0.33%
错误接受率6.17% / 0.83%
分类准确率94.67% / 99.33%

Table 4

Comparison of network architecture performance"

支路网络训练精度测试精度
AlexNet0.99290.9927
ResNet180.99590.9956
GoogLeNet0.99370.9938
VGG160.99630.9958
IVGG0.99870.9990

Table 5

Performance comparison with other algorithms"

测试方法准确率/%

FDCT-TA

NSST-EGLG

文献[3

文献[12

本文

92.34

94.27

98.66

93.30

99.62

1 刘恩泽, 吴文福. 基于机器视觉的农作物表面多特征决策融合病变判断算法[J]. 吉林大学学报: 工学版, 2018, 48(6): 1873-1878.
Liu En-ze, Wu Wen-fu. Agricultural surface multiple feature decision fusion diseasejudgment algorithm based on machine vision[J]. Journal of Jilin University(Engineering and Technology Edition), 2018, 48(6):1873-1878.
2 Zhang K H, Wang W D, Lv Z Q, et al. Computer vision detection of foreign objects in coal processing using attention CNN[J]. Engineering Applications of Artificial Intelligence, 2021,102: 104242-104253.
3 谢舰, 姚剑敏, 严群, 等. 基于深度学习的磁瓦表面缺陷分割与识别[J]. 液晶与显示, 2021, 36(5): 713-722.
Xie Jian, Yao Jian-min, Yan Qun, et al. Segmentation and recognition of magnetic tile surfacedefects based on deep learning[J]. Chinese Journal of Liquid Crystals and Displays, 2021, 36(5): 713-722.
4 Woo S, Park J, Lee J Y, et al. CBAM: convolutional block attention module[C]∥The 15th European Conference on Computer Vision, Munich, Germany, 2018: 3-19.
5 赵宏伟, 刘晓涵, 张媛, 等. 基于关键点注意力和通道注意力的服装分类算法[J]. 吉林大学学报: 工学版, 2020, 50(5): 1765-1770.
Zhao Hong-wei, Liu Xiao-han, Zhang Yuan, et al. Clothing classification algorithm based on landmark attention and channel attention[J]. Journal of Jilin University(Engineering and Technology Edition), 2020, 50(5): 1765-1770.
6 Qiu X J. A new multilevel feature fusion network for medical image segmentation[J]. Sensing and Imaging, 2021, 22(1): 1-20.
7 Haghighat M, Abdel-Mottaleb M, Alhalabi W. Discriminant correlation analysis: real-time feature level fusion for multimodal biometric recognition[J]. IEEE Transactions on Information Forensics and Security, 2016, 11(9): 1984-1996.
8 胡锦丽, 杨辉华, 刘振丙. 基于判别相关分析法的手指多模态融合身份识别方法[J]. 桂林电子科技大学学报, 2017, 37(6): 431-436.
Hu Jin-li, Yang Hui-hua, Liu Zhen-bing.A finger multi-modal fusion identification method based on discriminant correlation analysis[J]. Journal of Guilin University of Electronic Technology, 2017, 37(6): 431-436.
9 Pan H, Jiang J, Chen G. TDFSSD: top-down feature fusion single shot MultiBox detector[J].Signal Processing Image Communication, 2020, 89(4): 115987-115996.
10 Li X Q, Jiang H H, Yin G F. Detection of surface crack defects on ferrite magnetic tile[J]. NDT&E Int. 2014, 62: 6-13.
11 Yang C L, Liu P Y, Yin G F, et al. Crack detection in magnetic tile images using nonsubsampled shearlet transform and envelope gray level gradient[J]. Optics and Laser Technology, 2017, 90: 7-17.
12 刘畅, 张剑, 林建平. 基于神经网络的磁瓦表面缺陷检测识别[J]. 表面技术, 2019, 48(8): 330-339.
Liu Chang, Zhang Jian, Lin Jian-ping. Detection and identification of surface defects of magnetic tile based on neural network[J]. Surface Technology, 2019, 48(8): 330-339.
13 Huang Y B, Qiu C Y, Yuan K. Surface defect saliency of magnetic tile[J]. The Visual Computer, 2020, 36(1): 85-96.
[1] Zhen-hai ZHANG,Kun JI,Jian-wu DANG. Crack identification method for bridge based on BCEM model [J]. Journal of Jilin University(Engineering and Technology Edition), 2023, 53(5): 1418-1426.
[2] Ze-qiang ZHANG,Wei LIANG,Meng-ke XIE,Hong-bin ZHENG. Elite differential evolution algorithm for mixed⁃model two⁃side disassembly line balancing problem [J]. Journal of Jilin University(Engineering and Technology Edition), 2023, 53(5): 1297-1304.
[3] Peng YU,Yan PIAO. New method for extracting person re-identification attributes based on multi-scale features [J]. Journal of Jilin University(Engineering and Technology Edition), 2023, 53(4): 1155-1162.
[4] Yu JIANG,Jia-zheng PAN,He-huai CHEN,Ling-zhi FU,Hong QI. Segmentation-based detector for traditional Chinese newspaper [J]. Journal of Jilin University(Engineering and Technology Edition), 2023, 53(4): 1146-1154.
[5] Zhen-yu WU,Xiao-fei LIU,Yi-pu WANG. Trajectory planning of unmanned system based on DKRRT*⁃APF algorithm [J]. Journal of Jilin University(Engineering and Technology Edition), 2023, 53(3): 781-791.
[6] Bo TAO,Fu-wu YAN,Zhi-shuai YIN,Dong-mei WU. 3D object detection based on high⁃precision map enhancement [J]. Journal of Jilin University(Engineering and Technology Edition), 2023, 53(3): 802-809.
[7] Shan XUE,Ya-liang ZHANG,Qiong-ying LYU,Guo-hua CAO. Anti⁃unmanned aerial vehicle system object detection algorithm under complex background [J]. Journal of Jilin University(Engineering and Technology Edition), 2023, 53(3): 891-901.
[8] Hong-yang PAN,Zhao LIU,Bo YANG,Geng SUN,Yan-heng LIU. Overview of swarm intelligence methods for unmanned aerial vehicle systems based on new⁃generation information technology [J]. Journal of Jilin University(Engineering and Technology Edition), 2023, 53(3): 629-642.
[9] Ying HE,Jun-song FAN,Wei WANG,Geng SUN,Yan-heng LIU. Joint optimization of secure communication and trajectory planning in unmanned aerial vehicle air⁃to⁃ground [J]. Journal of Jilin University(Engineering and Technology Edition), 2023, 53(3): 913-922.
[10] Peng GUO,Wen-chao ZHAO,Kun LEI. Dual⁃resource constrained flexible job shop optimal scheduling based on an improved Jaya algorithm [J]. Journal of Jilin University(Engineering and Technology Edition), 2023, 53(2): 480-487.
[11] Jin-Zhen Liu,Guo-Hui Gao,Hui Xiong. Multi⁃scale attention network for brain tissue segmentation [J]. Journal of Jilin University(Engineering and Technology Edition), 2023, 53(2): 576-583.
[12] Xian-yu QI,Wei WANG,Lin WANG,Yu-fei ZHAO,Yan-peng DONG. Semantic topological map building with object semantic grid map [J]. Journal of Jilin University(Engineering and Technology Edition), 2023, 53(2): 569-575.
[13] Xiao-hu SHI,Jia-qi WU,Chun-guo WU,Shi CHENG,Xiao-hui WENG,Zhi-yong CHANG. Residual network based curve enhanced lane detection method [J]. Journal of Jilin University(Engineering and Technology Edition), 2023, 53(2): 584-592.
[14] Xuan-jing SHEN,Xue-feng ZHANG,Yu WANG,Yu-bo JIN. Multi⁃focus image fusion algorithm based on pixel⁃level convolutional neural network [J]. Journal of Jilin University(Engineering and Technology Edition), 2022, 52(8): 1857-1864.
[15] Fu-heng QU,Tian-yu DING,Yang LU,Yong YANG,Ya-ting HU. Fast image codeword search algorithm based on neighborhood similarity [J]. Journal of Jilin University(Engineering and Technology Edition), 2022, 52(8): 1865-1871.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!