Improved YOLOv5s Fabric Defect Detection Algorithm Based on the Fusion of Global Attention Mechanism and Decoupling Head

Shunqi Mei, Zhi Xiao, Jia Chen, Shiyi Shan and Shi Zhou
Additional contact information
Shunqi Mei: Wuhan Textile University, China
Zhi Xiao: Wuhan Textile University, China
Jia Chen: Wuhan Textile University, China
Shiyi Shan: Wuhan Textile University, China
Shi Zhou: Wuhan Textile University, China

International Journal of Information Retrieval Research (IJIRR), 2024, vol. 14, issue 1, 1-18

Abstract: During fabric production, machine malfunctions and yarn breakages can lead to defects, affecting product quality. Thus, fabric defect detection is crucial for quality control. Existing detection systems struggle with speed and accuracy, especially for small defects, due to the diversity of defect types and shapes. This paper introduces an improved YOLOv5s algorithm, YOLOv5s-GSD, which enhances detection performance by integrating a Global Attention Mechanism (GAM), the SIOU loss function, and a decoupling head. The GAM enhances focus on relevant features, improving detection accuracy; the SIOU loss function replaces the GIOU loss function to optimize the vector angle of regressions, enhancing convergence speed and accuracy; and the decoupling head separates classification and regression tasks, further improving detection performance. Experimental results show the improved algorithm achieves an accuracy of 97.6% and a recognition rate of 36 frames per second, effectively reducing the miss rate of small targets.

Date: 2024
References: Add references at CitEc
Citations:

Downloads: (external link)
http://services.igi-global.com/resolvedoi/resolve.aspx?doi=10.4018/IJIRR.356489 (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:igg:jirr00:v:14:y:2024:i:1:p:1-18

Access Statistics for this article

International Journal of Information Retrieval Research (IJIRR) is currently edited by Zhongyu Lu

More articles in International Journal of Information Retrieval Research (IJIRR) from IGI Global
Bibliographic data for series maintained by Journal Editor ().