Lightweight defect detection method of punched nickel-plated steel strip based on GhostNet

Aiming at the problem that the defects generated in the production and transportation of punched nickel-plated steel strips are not easy to be detected by deep learning methods, a lightweight, low-redundancy, and high-precision detection method is proposed in this paper. Firstly, a feature extraction network based on GhostNet is constructed, which reduces the amount of computation and feature redundancy while ensuring accuracy. Then the ECA module is applied to the detection head to perform weighted fusion of the features of different channels for better differentiation. Finally, the YOLO detection head is used for multi-scale detection. In the experiment, the mAP of 84.86% was obtained by this method, which proves that this method can be applied to the actual steel strip defect: detection.