ZFD-Net: Zinc flower defect detection model of galvanized steel surface based on improved YOLOV5.
Journal:
PloS one
Published Date:
Jan 1, 2025
Abstract
Due to the complex factory environment, zinc flower defects and galvanized sheet background are difficult to distinguish, and the production line running speed is fast, the existing detection methods are difficult to meet the needs of real-time detection in terms of accuracy and speed. We propose ZFD-Net, a zinc flower defect detection model on the surface of galvanized sheet based on improved you only look once (YOLO)v5. Firstly, the model combined the YOLOV5 model with our proposed cross stage partial transformer (CSTR) module in this paper to increase the model receptive field and improve the global feature extraction (FE) capability. Secondly, we use bi-directional feature pyramid network (Bi-FPN) weighted bidirectional feature pyramid network to fuse defect details of different levels and scales to improve them. Then we propose a cross resnet simam fasternet (CRSFN) module to improve the reasoning speed of ZFD-Net and ensure the detection effect of zinc flower defects. Finally, we construct a high-quality dataset of zinc flower defect (ZFD) detection on galvanized sheet surface, which solves the problem that no public dataset is available at present. ZFD-Net is compared with state-of-the-art (SOTA) methods on the self-built data set, and its performance indicators are better than all methods.
