An Industrial Micro Parts Recognition Technology Based on Improved Yolov8

Abstract Aiming at the recognition and classification problem of micro assembly parts in industrial production, the Yolov8 object detection algorithm was used as the base model to improve the C2f class in the backbone. And apply the improved object detection algorithm to the recognition field of detecting assembly parts. By collecting images of gears, nails, and screws from various angles on Roboflow, a training set and a validation set containing three types of gears, nails, and screws are formed. At the same time, in order to improve the robustness of the Yolov8 object detection algorithm, data processing methods such as adding noise and image rotation were adopted for the samples in the data. Due to the use of Yolov8 as the base model, a TXT text file containing label prediction box position information was formed after labelling the dataset on Roboflow. Based on Python (3.10.12) and cloud platform, improved Yolov8 object detection algorithm was trained and network validation was carried out. Use TXT files as input for the model to obtain and learn features of gears, nails, and screws. After 200 epochs, the mAP for detecting gears, nails, and screws increased by about 3% compared to the base model.