# ZFD-Net: Zinc flower defect detection model of galvanized steel surface based on improved YOLOV5

## TL;DR

This paper introduces ZFD-Net, an improved YOLOv5 model for detecting zinc flower defects on galvanized steel surfaces, achieving better accuracy and speed in real-time industrial settings.

## Contribution

The novel ZFD-Net model integrates a CSTR module, Bi-FPN, and CRSFN to enhance detection accuracy and speed for zinc flower defects.

## Key findings

- ZFD-Net outperforms state-of-the-art methods in detecting zinc flower defects on galvanized steel surfaces.
- The proposed model improves global feature extraction and defect detail fusion for better accuracy.
- A high-quality ZFD dataset was constructed to address the lack of public datasets for this task.

## 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.