Computer Science ›› 2023, Vol. 50 ›› Issue (11A): 230100100-7.doi: 10.11896/jsjkx.230100100

• Image Processing & Multimedia Technology • Previous Articles     Next Articles

Magnetic Tile Defect Detection Algorithm Based on Improved YOLOv4

ZHANG Xiaoxiao, DENG Chengzhi, WU Zhaoming, CAO Chunyang, HU Cheng   

  1. School of Information Engineering,Nanchang Institute of Technology,Nanchang 330099,China
  • Published:2023-11-09
  • About author:ZHANG Xiaoxiao,born in 1992,MS candidate.Her main research interests include deep learning and computer vision.
    DENG Chengzhi,born in 1964,Ph.D,professor,master supervisor.His main research interests include RS-image processing and image processing.
  • Supported by:
    Jiangxi Province Postgraduate Innovation Special Fund Project(YC2021-S184) and Nanchang Institute of Tech-nology Postgraduate Innovation Special Fund Project(YJSCX202130).

Abstract: Various defects occur in the manufacturing process of magnetic tiledue to process problems,and traditional detection algorithms have slow detection speed and low accuracy.In order to achieve fast and effective detection of surface defects of magnetic tiles,this paper proposes a defect detection method for magnetic tiles with improved YOLOv4 algorithm.Firstly,the scSE attention module is embedded in the residual unit of CSPnet in the feature extraction backbone network to enhance the spatial features and channel features of small targets.Secondly,the empty convolutional space pooling pyramid(ASPP) module is used instead of the original SPP module to increase the perceptual field of convolutional kernel,retain more image details and enhance information relevance.Finally,the traditional convolution in the five convolution blocks is replaced by the depth-separable convolution in the neck part to better extract the feature information and reduce the number of parameters of the model.Experimental results show that the improved YOLOv4 algorithm achieves an average accuracy value of 96.67%,a detection speed of 44 ms,and a model size of 249 MB.It is significantly better than the original algorithm and has higher detection accuracy and practicality.

Key words: Defect detection, YOLOv4, scSE attention, Void convolution pooling, Depth-separabl

CLC Number: 

  • TP391
[1]LIU C,ZHANG J,LIN J P.Detection and Identification of Surface Defects of Magnetic Tile Based on Neural Network[J].Surface Technology,2019,48(8):330-339.
[2]BHARATI P,PRAMANIK A.Deep learning techniques-R-CNN to mask R-CNN:a survey[C]//Computational Intelligence in Pattern Recognition(CIPR 2019).2020:657-668.
[3]GIRSHICK R.Fast r-cnn[C]//Proceedings of the IEEE International Conference on Computer Vision.2015:1440-1448.
[4]CHEN Y,LI W,SAKARIDIS C,et al.Domain adaptive faster r-cnn for object detection in the wild[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.2018:3339-3348.
[5]JIANG P,ERGU D,LIU F,et al.A Review of Yolo algorithm developments[J].Procedia Computer Science,2022,199:1066-1073.
[6]LOEY M,MANOGARAN G,TAHA M H N,et al.Fightingagainst COVID-19:A novel deep learning model based on YOLO-v2 with ResNet-50 for medical face mask detection[J].Sustainable Cities and Society,2021,65:102600.
[7]CAO X,ZHANG F,YIC,et al.Wafer Surface Defect Detection Based On Improved YOLOv3 Network[C]//2020 5th International Conference on Mechanical,Control and Computer Engineering (ICMCCE).2020.
[8]BOCHKOVSKIY A,WANG C Y,LIAO H Y M.Yolov4:Optimal speed and accuracy of object detection[J].arXiv:2004.10934,2020.
[9]WANG P,SHI H,YANG X,et al.Three-way k-means:integrating k-means and three-way decision[J].International Journal of Machine Learning and Cybernetics,2019,10:2767-2777.
[10]XIE H,LI Y,LI X,et al.A Method for Surface Defect Detection of Printed Circuit Board Based on Improved YOLOv4[C]//2021 IEEE 2nd International Conference on Big Data,Artificial Intelligence and Internet of Things Engineering (ICBAIE).IEEE,2021.
[11]CHEN Q,WANG P S,CHENG A D,et al.Robust one-stage object detection with location-aware classifiers[J].Pattern Recognition,2020,105.
[12]WANG C Y,BOCHKOVSKIY A,LIAO H Y M.Scaled-yolov4:Scaling cross stage partial network[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition.2021:13029-13038.
[13]LI G,SUO R,ZHAO G,et al.Real-time detection of kiwifruit flower and bud simultaneously in orchard using YOLOv4 for robotic pollination[J].Computers and Electronics in Agriculture,2022,193:106641.
[14]WANG C Y,LIAO H Y M,WU Y H,et al.CSPNet:A new backbone that can enhance learning capability of CNN[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops.2020:390-391.
[15]HE K,ZHANG X,REN S,et al.Spatial pyramid pooling in deep convolutional networks for visual recognition[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2015,37(9):1904-1916.
[16]LIU S,QI L,QIN H,et al.Path aggregation network for instance segmentation[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.2018:8759-8768.
[17]GHIASI G,LIN T Y,LE Q V.Nas-fpn:Learning scalable feature pyramid architecture for object detection[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition.2019:7036-7045.
[18]OBESO A M,BENOIS-PINEAU J,V?ZQUEZ M S G,et al.Visual vs internal attention mechanisms in deep neural networks for image classification and object de-tection[J].Pattern Recognition,2022,123:108411.
[19]KUMAR A,SINGH SODHI S.Classification of data on stackedautoencoder using modified sigmoid activation function[J].Journal of Intelligent & Fuzzy Systems,2023(Preprint):1-18.
[20]HANANDEH S,ARDAH A,ABU-FARSAKH M.Using artificial neural network and ge-netics algorithm to estimate the resi-lient modulus for stabilized subgrade and propose new empirical formula[J].Transportation Geotechnics,2020,24:100358.
[21]HUANG Y,QIU C,YUAN K.Surface defect saliency of magnetic tile[J].The Visual Computer,2020,36:85-96.
[1] BAI Mingli, WANG Mingwen. Fabric Defect Detection Algorithm Based on Improved Cascade R-CNN [J]. Computer Science, 2023, 50(6A): 220300224-6.
[2] JIANG Bo, WAN Yi, XIE Xianzhong. Improved YOLOv5s Lightweight Steel Surface Defect Detection Model [J]. Computer Science, 2023, 50(11A): 230900113-7.
[3] LUO Yuetong, LI Chao, DUAN Chang, ZHOU Bo. Hue Augmentation Method for Industrial Product Surface Defect Images [J]. Computer Science, 2023, 50(11A): 230200089-6.
[4] ZOU Chenwei, YAO Rao. Study on Ultrasonic Phased Array Defect Detection Based on Machine Vision [J]. Computer Science, 2023, 50(11A): 230200150-6.
[5] CHU Yu-chun, GONG Hang, Wang Xue-fang, LIU Pei-shun. Study on Knowledge Distillation of Target Detection Algorithm Based on YOLOv4 [J]. Computer Science, 2022, 49(6A): 337-344.
[6] LI Fa-guang, YILIHAMU·Yaermaimaiti. Real-time Detection Model of Insulator Defect Based on Improved CenterNet [J]. Computer Science, 2022, 49(5): 84-91.
[7] ZHAO Chen-yang, ZHANG Hui, LIAO De, LI Chen. Rail Surface Defect Detection Model Based on Attention Module and Hybrid-supervised Learning [J]. Computer Science, 2022, 49(11A): 210800241-6.
[8] FU Bo-wen, LI Chuang-chuang, LIANG Ai-hua. Facial Landmark Fast Detection Based on Improved YOLOv4-tiny [J]. Computer Science, 2022, 49(11A): 211100290-5.
[9] PENG Lei, ZHANG Hui. U-net for Pavement Crack Detection [J]. Computer Science, 2021, 48(11A): 616-619.
[10] XIE Yuan, MIAO Yu-bin, XU Feng-lin, ZHANG Ming. Injection-molded Bottle Defect Detection Using Semi-supervised Deep Convolutional Generative Adversarial Network [J]. Computer Science, 2020, 47(7): 92-96.
[11] LUO Yue-tong,BIAN Jing-shuai,ZHANG Meng,RAO Yong-ming,YAN Feng. Detection Method of Chip Surface Weak Defect Based on Convolution Denoising Auto-encoders [J]. Computer Science, 2020, 47(2): 118-125.
[12] HU Hai-bing, XU Ting, ZHANG Bo, XU Dong-jian, JIN Shi-qun, LU Rong-sheng. Application of Open MP and Ring Buffer Technology in Defects Detection of Glass Substrate [J]. Computer Science, 2019, 46(11A): 562-566.
[13] SU Ruo, WU Ji, LIU Chao, YANG Hai-yan. Requirement Defect Detection Based on Multi-view Card Model [J]. Computer Science, 2018, 45(10): 183-188.
[14] ZHU Chao-yang, CHEN Xiang-zhou, YAN Long and ZHANG Xin-ming. Research on Software Defect Prediction Based on AIRS Using PCA [J]. Computer Science, 2017, 44(Z6): 483-485.
[15] QIAO Li, ZHAO Er-dun, LIU Jun-jie and CHENG Bin. Research of Workpiece Defect Detection Method Based on CNN [J]. Computer Science, 2017, 44(Z11): 238-243.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!