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

Yuntao Xu, Peigang Jiao and Jiaqi Liu

PLOS ONE, 2023, vol. 18, issue 12, 1-16

Abstract: 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.

Date: 2023
References: View complete reference list from CitEc
Citations: View citations in EconPapers (1)

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0289179 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 89179&type=printable (application/pdf)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:plo:pone00:0289179

DOI: 10.1371/journal.pone.0289179

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().