Valid-IoU: an improved IoU-based loss function and its application to detection of defects on printed circuit boards

Elnaz Vakili,Ghader Karimian,Maryam Shoaran,Reza Yadipour,Jafar Sobhi
DOI: https://doi.org/10.1007/s11042-024-19482-4
IF: 2.577
2024-06-14
Multimedia Tools and Applications
Abstract:The printed circuit boards (PCBs) should be inspected during the manufacturing process to minimize defects such as printing errors, incorrect component selections, and incorrect soldering. Convolutional neural networks (CNNs) have emerged as a powerful tool for this purpose, owing to their robust recognition capabilities. However, the accuracy of CNNs is significantly influenced by the choice of loss function. Intersection over Union (IoU) based loss functions commonly employed for bounding box regression, achieve remarkable success, but still have some main drawbacks such as inaccurate predicting bounding box dimensions relative to ground truth data. In this paper, we propose an improved IoU called Valid-IoU (VIoU) to solve this problem. We integrated the VIoU loss into state-of-the-art YOLO object detection frameworks to show its effectiveness in PCB defect detection. The experimental results on the HRIPCB PCB dataset demonstrate that the VIoU loss has high defect detection accuracy. Using YOLOv4 helped us to improve average accuracy, and with YOLOv8, we achieved faster detection speed compared to other methods.
computer science, information systems, theory & methods,engineering, electrical & electronic, software engineering
What problem does this paper attempt to address?