YOLOv5s fabric defect detection model with mixed attention mechanism

Abstract

In the field of industrial quality inspection, fabric defect detection is regarded as an important research topic. The traditional inspection method mainly relies on manual visual inspection, which is time-consuming and easy to be affected by human factors, and has the problems of low accuracy, slow speed and high cost. Therefore, it is very important to use digital image processing technology and target detection technology to develop high precision, high efficiency and low-cost fabric defect detection model suitable for application deployment. Driven by deep learning-based neural network technology, many models with excellent performance have emerged in the field of object detection. This paper will improve on the basis of YOLOv5s model. ACmix module is introduced in Backbone layer of YOLOv5s model. This module overcomes the problem that convolution operation may cause local information loss after feature extraction, improves the model's limitation on feature extraction of complex background images, and enhances the model's processing ability in local feature fusion. Compared with the original YOLOv5s algorithm, mAP has improved by 0.6%. In the complex background, the detection accuracy of fabric defects targets has been significantly optimized and improved.