An Improved YOLOv9 and Its Applications for Detecting Flexible Circuit Boards Connectors

Autor: Gengjie Huang, Yinbing Huang, Haoyang Li, Ziwen Guan, Xuecong Li, Guidong Zhang, Wendong Li, Xiran Zheng
Jazyk: angličtina
Rok vydání: 2024
Předmět:
Zdroj: International Journal of Computational Intelligence Systems, Vol 17, Iss 1, Pp 1-12 (2024)
Druh dokumentu: article
ISSN: 1875-6883
DOI: 10.1007/s44196-024-00669-4
Popis: Abstract Flexible circuit boards are a cornerstone of the modern electronics industry. In automatic defect detection, FPC connectors present challenges such as minimal differences between oxidation defects and the background, easy degradation of Intersection over Union (IoU) scores, and significant variations in the shapes of black defect boundaries. Consequently, existing algorithms perform poorly in this task. We improve model YOLOv9 by introducing Multi-scale Dilated Attention (MSDA) on the output side to enhance the ability to capture features, and Deformable Large Kernel Attention (DLKA) on the other side of the output header to improve the ability to adapt to complex defect boundaries. Our use of IoU loss completely eliminates the risk of IoU degradation or gradient vanishing. Furthermore, we reduce computational overhead with the implementation of Faster Block. Following these improvements, the mean Average Precision (mAP) at 75% IoU (mAP75) for oxidized defects increased by 7.5% relative to the base model. Similarly, the mAP at 50% IoU (mAP50) for black defects increased by 5.7%, validating the relevance and efficacy of our proposed improvements. Overall, the average mAP50, mAP75, and mAP50:95 for all defects improved by 3.8%, 2.0%, and 2.3%, respectively. The performance gain achieved by our enhanced model significantly exceeds the improvement of YOLOv9 relative to YOLOv8.
Databáze: Directory of Open Access Journals