STD2: Swin Transformer-Based Defect Detector for Surface Anomaly Detection

Abstract

In the quest for manufacturing excellence, this article presents a pioneering approach to defect detection on steel surfaces. Complex and small defects on steel surfaces manifest in diverse forms, such as irregular shapes and varying sizes, making it challenging to devise a singular detection model capable of effectively detecting these multifarious defect types. Yet, another challenge lies in the close similarity between defects and nondefective components. Additionally, the fluctuating environmental factors exposed to steel structures introduce noise and artifacts into image data, compounding the complexity of precise defect detection. To address these issues, we initially employed the Swin Transformer as the core element of Faster R-CNN. Second, we ingeniously integrated a path aggregation feature pyramid network (PAFPN) into the architecture. This allows us to capture a comprehensive array of defect feature maps and enhance the network’s capability to identify defects at various scales. After that, we swapped out the region-of-interest (RoI) pooling for the more sophisticated deformable pooling method to get more precise defect localization information. Finally, the standard intersection over union (IoU) loss is substituted with complete intersection over union (CIoU) to address issues of duplicate proposals and improve overall performance by taking into account the geometry of the bounding boxes in addition to their overlaps. Our proposed Swin Transformer Defect Detector (STD2) has been rigorously evaluated on diverse steel surface defect datasets, including NEU-DET (yielding an impressive mean average precision (mAP) of 81.05%), GC10-DET (with a notable mAP score of 72.38%), and printed circuit board (PCB) (achieving an outstanding mAP of 99.00%), which demonstrate its exceptional performance in surface anomaly detection. The code is available at <uri xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">https://github.com/Shuvo001/STD2</uri>.