Visual recognition of wheel hubs with convolutional neural network

Abstract

The traditional recognition methods of wheel hubs are mainly based on extracted feature matching. In practical production, their accuracy, robustness and processing speed are usually greatly affected. To overcome these problems, this paper proposes a recognition method based on convolutional neural network. The basic steps include two parts: wheel image pre-processing and wheel model classification. The image processing method, mainly using the detection algorithm of hough circles, obtains the center coordinates and radius of the wheel. Then it maps the ring-shaped wheel in right-angle coordinates to polar coordinates by the center coordinates and radius. This stepcan extract the ring-shaped feature information of the wheel image and reduce the influence generated by redundant features. Then a network architecture with an improved Resnet is designed to classify the wheel models. Finally, the wheel model recognition algorithm is evaluated, and the effectiveness of the method is verified through the comparison experiments of SVM, KNN and other models. The experiments show that the recognition accuracy can reach about 99.8% for 10 kinds of wheels.