A strip steel surface defect detection method based on attention mechanism and multi-scale maxpooling

In industry, defect detection involves two kinds of tasks: defect classification and location, which make it difficult to ensure the accuracy of both, and also make the task still challenging in practical application. Based on the analysis of the advantages and disadvantages of the current defect detection method, this paper proposes a defect detection method based on attention mechanism and multi-scale maxpooling (MSMP). In order to effectively improve the detection accuracy of the model, we use Resnet50 as the pre-training network construct two-stage detection model which is used to be the baseline network, and introduce the attention mechanism and MSMP module on this basis. The attention mechanism can enhance the features of the feature map extracted in each stage of Resnet50, so that the network concentrates on the effective areas for the final detection results, and ignores the background areas that are invalid or even unfavorable for detection. The proposed MSMP can incrementally enhance the receptive field, distinguish the most significant context features, and effectively improve detection precision. The proposed method is used to train and test on the NEU-DET dataset. Compared with the baseline network without any improvement, the proposed method in this paper achieves 3.65% mAP performance improvement. Meanwhile, our method achieves a performance improvement of 3.65% mAP. In addition, compared with the feature fusion mechanism, our method improves 4.03% mAP. Moreover, compared with the attention mechanisms such as spatial attention and SE block, our method improves 1.51%/1.03% mAP. Furthermore, compared with the one-stage detection algorithm SSD/YOLO-V4, the proposed method improves 5.01%/4.92% mAP. In addition, the classification accuracy of our model is as high as 94.73%.

Automated summary

The study proposes a defect detection method based on attention mechanism and multi-scale maxpooling, using Resnet50 to construct a two-stage detection model. The method was trained and tested on the NEU-DET dataset, achieving a 3.65% mAP performance improvement compared to the baseline network. Additionally, the classification accuracy of the method reaches as high as 94.73%.