Scratch Detection Method of Shaft Parts Based on Improved YOLOv5

Haodong Bai,Guanyu Chen,Huiting Fu,Changying Liu
DOI: https://doi.org/10.1088/1742-6596/2560/1/012024
2023-08-26
Journal of Physics: Conference Series
Abstract:The pre-extracted target object is always distorted due to many parts spots, surface reflection and other factors in traditional visual inspection. Therefore, a detection process based on improved YOLOv5 is proposed. In the early stage, for the optimization of light source, the circular light is generally used to successfully extract the region of interest, so as to reduce the scratches caused by image pretreatment, resulting in detection omission and distortion. Adding the ACON (All Class in One Network, ACON) activation function can improve the network feature extraction ability and the accuracy of the detection algorithm. Bounding box loss, means of target detection loss and classification loss are all significantly improved by the improved algorithm, far exceeding the performance of traditional detection methods, and all have a certain degree of reduction.
What problem does this paper attempt to address?