Surface Defect Detection Algorithm of Hot-Rolled Strip Based on Improved YOLOv7

To enhance the capability of classifying and localizing defects on the surface of hot-rolled strips, this paper proposed an algorithm based on YOLOv7 to improve defect detection. The BI-SPPFCSPC structure was incorporated into the feature pyramid in this algorithm, enabling enhanced extraction of features from small objects and improved accuracy in network model positioning. Additionally, a small object detection layer was introduced to enhance shallow feature capture. Then the CARAFE sampling operator was used for up-sampling to reduce the feature loss problem of small objects. Finally, the WIoU served as the loss function for network model training to expedite convergence. The NEU-DET dataset was utilized for ablation and comparison tests. The findings indicated that the enhanced YOLOv7 model's mAP value had increased to 80.7%. The detection impact was much enhanced in comparison to other traditional models, and the frequency of false and missing detections was also decreased. © (2024), (International Association of Engineers). All Rights Reserved.