RC-YOLOv5s: for tile surface defect detection

被引:14
作者
Hou, Wenqing [1 ]
Jing, Huicheng [1 ]
机构
[1] North China Univ Sci & Technol, Tangshan 063000, Peoples R China
关键词
Deep learning; Defect detection; Tiles; RC-YOLOv5s; Attention mechanism;
D O I
10.1007/s00371-023-02793-2
中图分类号
TP31 [计算机软件];
学科分类号
081202 ; 0835 ;
摘要
To solve the problems of complex surface texture of magnetic tile, cumbersome process of traditional detection algorithm and low detection accuracy, this paper proposes a deep learning-based detection model: RC-YOLOv5s. The model incorporates two new structures: Res-Head and Drop-CA, where Res-Head enhances the feature fusion and information exchange between different layer structures and Drop-CA alleviates the case that the model pays too much attention to the defect target and reduces the missed detection rate of the model. Compared with YOLOv5s, the detection accuracy of the proposed model is improved by 1.83%, the missing rate is reduced from 1.673 to 0.372%, and the average detection frame rate of a single image reaches 41.67, which meets the requirements of real-time and accuracy of tile surface detection.
引用
收藏
页码:459 / 470
页数:12
相关论文
共 28 条
  • [1] Image-Based Surface Defect Detection Using Deep Learning: A Review
    Bhatt, Prahar M.
    Malhan, Rishi K.
    Rajendran, Pradeep
    Shah, Brual C.
    Thakar, Shantanu
    Yoon, Yeo Jung
    Gupta, Satyandra K.
    [J]. JOURNAL OF COMPUTING AND INFORMATION SCIENCE IN ENGINEERING, 2021, 21 (04)
  • [2] Bochkovskiy A, 2020, Arxiv, DOI arXiv:2004.10934
  • [3] Bi-deformation-UNet: recombination of differential channels for printed surface defect detection
    Chen, Ziyang
    Huang, Guoheng
    Wang, Ying
    Qiu, Junhao
    Yang, Fan
    Yu, Zhiwen
    Pun, Chi-Man
    Ling, Wing-Kuen
    [J]. VISUAL COMPUTER, 2023, 39 (09) : 3995 - 4013
  • [4] Dosovitskiy A, 2021, Arxiv, DOI arXiv:2010.11929
  • [5] MSFT-YOLO: Improved YOLOv5 Based on Transformer for Detecting Defects of Steel Surface
    Guo, Zexuan
    Wang, Chensheng
    Yang, Guang
    Huang, Zeyuan
    Li, Guo
    [J]. SENSORS, 2022, 22 (09)
  • [6] Deep Residual Learning for Image Recognition
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 770 - 778
  • [7] Coordinate Attention for Efficient Mobile Network Design
    Hou, Qibin
    Zhou, Daquan
    Feng, Jiashi
    [J]. 2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 13708 - 13717
  • [8] Searching for MobileNetV3
    Howard, Andrew
    Sandler, Mark
    Chu, Grace
    Chen, Liang-Chieh
    Chen, Bo
    Tan, Mingxing
    Wang, Weijun
    Zhu, Yukun
    Pang, Ruoming
    Vasudevan, Vijay
    Le, Quoc V.
    Adam, Hartwig
    [J]. 2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, : 1314 - 1324
  • [9] Detection of PCB Surface Defects With Improved Faster-RCNN and Feature Pyramid Network
    Hu, Bing
    Wang, Jianhui
    [J]. IEEE ACCESS, 2020, 8 : 108335 - 108345
  • [10] Hu H., 2019, Electromech. Eng., V36, P117