A Component Recognition Model Based on Improved Attention Mechanism and YOLOv3
DaiLin Jiang,Yikai Wu,Pengcheng Xu,Jie Yang,Delan Zhang,Jiawei Liu,Zhengquan Lyu
DOI: https://doi.org/10.1117/12.3026313
2024-01-01
Abstract:In response to the insufficient feature information of smaller electronic components in the current YOLOv3 algorithm used in object detection, which leads to false detections, missed detections, and low average accuracy, this paper proposes an improved attention mechanism-based YOLOv3 algorithm for small object detection. Firstly, the model introduces a category weighting factor α to overcome the problem of imbalanced training samples, enhancing training robustness and improving classification accuracy. Secondly, an attention mechanism is introduced to learn the importance of each channel's characteristics, enabling weight allocation for primary and secondary information, suppressing secondary information, and eliminating some redundant information, effectively improving the learning direction of the model. Lastly, by incorporating the category weighting factor α and L2 regularization term into the loss function, the training of the network model can converge faster and more stably. The proposed algorithm achieves an average precision of 92.86% for resistor recognition, a 1.07% improvement compared to the classical YOLOv3 algorithm. For chip recognition, the average precision is 97.23%, an improvement of 0.8% compared to the classical YOLOv3 algorithm. These results demonstrate a certain enhancement in the detection performance of small objects.