Detection of Ceramic Tile Surface Defects Based on Improved YOLOv5
CSTR:
Author:
  • Article
  • | |
  • Metrics
  • |
  • Reference [26]
  • |
  • Related [20]
  • | | |
  • Comments
    Abstract:

    The existing detection method of ceramic tile surface defects has the problem of insufficient ability to identify small target defects, and the detection speed needs to be improved. Therefore, this study proposes a ceramic tile surface defect detection method based on improved YOLOv5. Firstly, due to the small size of ceramic tile surface defects, the detection abilities of three target detection head branches of YOLOv5s are compared and analyzed. It is found that the effectiveness of the model that removes the large target detection head and retains only the medium and small target detection heads is optimal. Secondly, to further realize the lightweight of the model, the study applies ghost convolution and C3Ghost modules to replace the ordinary convolution and C3 modules of YOLOv5s in the Backbone network, thus reducing the number of model parameters and the calculation amount. Finally, the coordinate attention mechanism module is added at the end of the Backbone and Neck networks of YOLOv5s to solve the problem of no attention preference in the original model. The proposed method is tested on the Tianchi ceramic tile defect detection dataset. The results show that the mean precision of the improved detection model averages 66%, which is 1.8% higher than the original YOLOv5s model. Besides, the size of the model is only 10.14 MB, and the number of parameters and the calculation amount is reduced by 48.7% and 38.7% respectively compared with the original model.

    Reference
    [1] Sameer Ahamad N, Bhaskara Rao J. Analysis and detection of surface defects in ceramic tile using image processing techniques. Microelectronics, Electromagnetics and Telecommunications. New Delhi: Springer, 2016. 575–582.
    [2] Samarawickrama YC, Wickramasinghe CD. Matlab based automated surface defect detection system for ceremic tiles using image processing. Proceedings of the 6th National Conference on Technology and Management (NCTM). Malabe: IEEE, 2017. 34–39.
    [3] Alamsyah R, Wiranata AD, Rafie. Defect detection of ceramic tiles using median filtering, morphological techniques, gray level co-occurrence matrix and K-nearest neighbor method. Scientific Research Journal, 2019, 7(4): 41–46
    [4] Birlutiu A, Burlacu A, Kadar M, et al. Defect detection in porcelain industry based on deep learning techniques. Proceedings of the 19th International Symposium on Symbolic and Numeric Algorithms for Scientific Computing (SYNASC). Timisoara: IEEE, 2017. 263–270.
    [5] Girshick R, Donahue J, Darrell T, et al. Rich feature hierarchies for accurate object detection and semantic segmentation. Proceedings of the 2014 IEEE Conference on Computer Vision and Pattern Recognition. Columbus: IEEE, 2014. 580–587.
    [6] Girshick R. Fast R-CNN. Proceedings of the 2015 IEEE International Conference on Computer Vision. Santiago: IEEE, 2015. 1440–1448.
    [7] Ren SQ, He KM, Girshick R, et al. Faster R-CNN: Towards real-time object detection with region proposal networks. Proceedings of the 28th International Conference on Neural Information Processing Systems. Montreal: MIT Press, 2015. 91–99.
    [8] Cai ZW, Vasconcelos N. Cascade R-CNN: Delving into high quality object detection. Proceedings of the 2018 IEEE Conference on Computer Vision and Pattern Recognition. Salt Lake City: IEEE, 2018. 6154–6162.
    [9] Redmon J, Divvala S, Girshick R, et al. You only look once: Unified, real-time object detection. Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition. Las Vegas: IEEE, 2016. 779–788.
    [10] Liu W, Anguelov D, Erhan D, et al. SSD: Single shot multibox detector. Proceedings of the 14th European Conference on Computer Vision. Amsterdam: Springer, 2016. 21–37.
    [11] Redmon J, Farhadi A. YOLOv3: An incremental improvement. arXiv:1804.02767, 2018.
    [12] Bochkovskiy A, Wang CY, Liao HYM. YOLOv4: Optimal speed and accuracy of object detection. arXiv:2004.10934, 2020.
    [13] Lin TY, Goyal P, Girshick R, et al. Focal loss for dense object detection. Proceedings of the 2017 IEEE International Conference on Computer Vision. Venice: IEEE, 2017. 2980–2988.
    [14] Lu QH, Lin JM, Luo LF, et al. A supervised approach for automated surface defect detection in ceramic tile quality control. Advanced Engineering Informatics, 2022, 53: 101692. [doi: 10.1016/j.aei.2022.101692
    [15] Hu J, Shen L, Sun G. Squeeze-and-excitation networks. Proceedings of the 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition. Salt Lake City: IEEE, 2018. 7132–7141.
    [16] Han K, Wang YH, Tian Q, et al. GhostNet: More features from cheap operations. Proceedings of the 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition. Seattle: IEEE, 2020. 1577–1586.
    [17] Hou QB, Zhou DQ, Feng JS. Coordinate attention for efficient mobile network design. Proceedings of the 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition. Nashville: IEEE, 2021. 13708–13717.
    [18] He KM, Zhang XY, Ren SQ, et al. Spatial pyramid pooling in deep convolutional networks for visual recognition. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2015, 37(9): 1904–1916. [doi: 10.1109/TPAMI.2015.2389824
    [19] Liu S, Qi L, Qin HF, et al. Path aggregation network for instance segmentation. Proceedings of the 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition. Salt Lake City: IEEE, 2018. 8759–8768.
    [20] Zheng ZH, Wang P, Liu W, et al. Distance-IoU loss: Faster and better learning for bounding box regression. Proceedings of the 34th AAAI Conference on Artificial Intelligence, AAAI 2020, the 32nd Innovative Applications of Artificial Intelligence Conference, IAAI 2020, the 10th AAAI Symposium on Educational Advances in Artificial Intelligence. New York: AAAI Press, 2020: 12993–13000.
    [21] Rezatofighi H, Tsoi N, Gwak JY, et al. Generalized intersection over union: A metric and a loss for bounding box regression. Proceedings of the 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition. Long Beach: IEEE, 2019. 658–666.
    [22] Chollet F. Xception: Deep learning with depthwise separable convolutions. Proceedings of the 2017 IEEE Conference on Computer Vision and Pattern Recognition. Honolulu: IEEE, 2017. 1800–1807.
    [23] Tan MX, Le QV. EfficientNet: Rethinking model scaling for convolutional neural networks. Proceedings of the 36th International Conference on Machine Learning. Long Beach: PMLR, 2019. 6105–6114.
    [24] Zhang HY, Wang Y, Dayoub F, et al. VarifocalNet: An IoU-aware dense object detector. Proceedings of the 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition. Nashville: IEEE, 2021. 8510–8519.
    [25] He KM, Zhang XY, Ren SQ, et al. Deep residual learning for image recognition. Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition. Las Vegas: IEEE, 2016. 770–778.
    [26] Ge Z, Liu ST, Wang F, et al. YOLOX: Exceeding YOLO series in 2021. arXiv:2107.08430, 2021.
    Cited by
    Comments
    Comments
    分享到微博
    Submit
Get Citation

余松森,张明威,杨欢.改进YOLOv5的瓷砖表面缺陷检测.计算机系统应用,2023,32(8):151-161

Copy
Share
Article Metrics
  • Abstract:907
  • PDF: 2378
  • HTML: 2100
  • Cited by: 0
History
  • Received:January 19,2023
  • Revised:February 23,2023
  • Online: May 22,2023
Article QR Code
You are the first990577Visitors
Copyright: Institute of Software, Chinese Academy of Sciences Beijing ICP No. 05046678-3
Address:4# South Fourth Street, Zhongguancun,Haidian, Beijing,Postal Code:100190
Phone:010-62661041 Fax: Email:csa (a) iscas.ac.cn
Technical Support:Beijing Qinyun Technology Development Co., Ltd.

Beijing Public Network Security No. 11040202500063