Abstract
Automated surface defect localization is closely related to product quality control. Although many image segmentation models have achieved excellent results in industrial defect localization, the wide-ranging defects in magnetic tile materials and the similarity between textures and background textures make it difficult to balance detection model performance and model size. Therefore, we propose a new magnetic tile defect localization model called MT-U2Net. The model consists of three parts: Dres-Ublock, U2 structure, and attention mechanism. Dres-Ublock is used to enhance the model's feature extraction capability and reduce its size; the U2-Net structure is utilized to learn cross-scale defect information; the attention mechanism is employed to eliminate redundant information. MT-U2Net improves detection performance across various defect scales. In the test set, MT-U2Net achieves an F1 Score of 0.840 and an MAE of 0.024 and compresses the model size to 3.39 M. Extensive experimental results show that MT-U2Net outperforms existing state-of-the-art algorithms in magnetic tile defect detection performance and model size.
Published Version
Talk to us
Join us for a 30 min session where you can share your feedback and ask us any queries you have
Schedule a call