Detection of Surface Defects of Magnetic Tiles Based on Improved YOLOv5

Figure 3

Network architecture of improved YOLOv5. We added the CBAM in neck part and an extra detection layer in prediction part.