Details

Autor(en) / Beteiligte

• Xu, Yuntao
• Jiao, Peigang
• Liu, Jiaqi
• Bilal, Anas

Titel

CFM-YOLOv5:CFPNet moudle and muti-target prediction head incorporating YOLOv5 for metal surface defect detection

Ist Teil von

• PloS one, 2023-12, Vol.18 (12), p.e0289179-e0289179

Ort / Verlag

United States: Public Library of Science

Erscheinungsjahr

2023

Quelle

MEDLINE

Beschreibungen/Notizen

• Aiming at the problem of low efficiency of manual detection in the field of metal surface defect detection, a deep learning defect detection method based on improved YOLOv5 algorithm is proposed. Firstly, in the feature enhancement part, we replace the multi-head self-attention module of the standard transformer encoder with the EVC module to improve the feature extraction ability. Second, in the prediction part, adding a small target detection head can reduce the negative impact of drastic object scale changes and improve the accuracy and stability of detection. Finally, the performance of the algorithm is verified by ablation experiments and analogy experiments. The experimental results show that the improved algorithm has greatly improved mAP and FPS on the data set, and can quickly and accurately identify the types of metal surface defects, which has reference significance for practical industrial applications.