Research on tile surface defect detection by improved YOLOv5

Abstract

Traditional methods for detecting surface defects on ceramic tiles result in misdetection and missed detection, which makes it difficult to guarantee product stability and consistency within the same batch. Therefore, this article proposes an improved YOLOv5 algorithm for detecting surface defects on ceramic tiles. Firstly, the Res2Net module is combined with self-attention to fully utilize local and global information and improve the feature extraction effect of defects. Secondly, the GS-BiFPN neck network is designed to enhance the fusion capability of shallow detail and deep semantic information and alleviate ambiguity and redundancy on the feature map. Then, a lightweight attention module is introduced to improve the detection capability of difficult-to-recognize defects and anti-background interference. Finally, the SIoU loss function improves the model’s convergence speed and accuracy. Experimental results demonstrate that the improved algorithm’s mean average precision (mAP) reaches 73.3%, 6.3% higher than the baseline model. Even when compared with YOLOv7-tiny, the mAP of the improved algorithm has increased by 8.7%. Additionally, the detection speed of the model can reach 92 frames per second, which can meet the requirements of ceramic tile surface defect detection in industrial scenarios.