Research and Application of Multi-Objective Workpiece Classification and Recognition Based on YOLO v5 Model

Authors

  • Li Xia

DOI:

https://doi.org/10.62051/ijcsit.v4n3.40

Keywords:

Artifact recognition, YOLOV5, Deep learning

Abstract

Workpiece recognition is an important application of machine vision in industry. The article proposes a deep learning image recognition method for workpiece recognition and classification. First, the workpiece sample library is established, the model is built using the YOLOv5 algorithm, and the optimal model is obtained after training and parameter tuning; in the process of solving the model, it is tested for the workpiece dataset, and the results show that the model achieves an accuracy of more than 94% in the validation set, and finally, after the validation, the model can be accurately recognized and classified, and it provides effective decision-making support for the actual recognition of the workpiece.

Downloads

Download data is not yet available.

References

[1] Xue, Z.; Xu, R.; Bai, D.; Lin, H. YOLO-Tea: A Tea Disease Detection Model Improved by YOLOv5. Forests 2023, 14 (2), 415. https://doi.org/10.3390/ f14020415

[2] Xu, L.; Dong, S.; Wei, H.; Ren, Q.; Huang, J.; Liu, J. Defect Signal Intelligent Recognition of Weld Radiographs Based on YOLO V5-IMPROVEMENT. j. Manuf. . Process. 2023, 99, 373-381. https://doi.org/10.1016/j.jmapro.2023.05.058

[3] Wang, L.; Liu, X.; Ma, J.; Su, W.; Li, H. Real-Time Steel Surface Defect Detection with Improved Multi-Scale YOLO-V5. Processes 2023, 11 (5), 1357. https://doi.org/10.3390/pr11051357

[4] Taşyürek, M.; Öztürk, C. A Fine-Tuned YOLOv5 Deep Learning Approach for Real-Time House Number Detection. peerJ Comput. Sci. 2023, 9, e1453. https:// doi.org/10.7717/peerj-cs.1453

[5] Liu, Z.; Gao, X.; Wan, Y.; Wang, J.; Lyu, H. An Improved YOLOv5 Method for Small Object Detection in UAV Capture Scenes. IEEE Access 2023, 11, 14365 -14374. https://doi.org/10.1109/ACCESS.2023.3241005

[6] Aydin, B.; Singha, S. Drone Detection Using YOLOv5. Eng 2023, 4 (1), 416-433. https://doi.org/10.3390/eng4010025

Downloads

Published

21-12-2024

Issue

Section

Articles

How to Cite

Xia, L. (2024). Research and Application of Multi-Objective Workpiece Classification and Recognition Based on YOLO v5 Model. International Journal of Computer Science and Information Technology, 4(3), 360-366. https://doi.org/10.62051/ijcsit.v4n3.40