Research on mechanical parts classification and recognition technology based on improved convolutional neural network YOLOv5

Abstract

In the field of classification and recognition of mechanical parts, in order to solve the difficulties of small target detection, adhesion and overlapping parts classification and recognition, this paper proposes an automatic classification and recognition algorithm of mechanical parts based on improved convolutional neural network YOLOv5. The feature pyramid module PANet is added to the traditional YOLOv5 network structure to strengthen the characterization ability of the network. At the same time, NMS algorithm and image segmentation algorithm are used to improve the detection ability of network boundary segmentation, to improve the detection accuracy and recall rate. The experimental results show that both the traditional YOLOv5 and the improved YOLOv5 in this paper can realize the classification and identification of mechanical parts, and the improved YOLOv5 has better detection precision and accuracy, faster detection speed, and better network robustness.