The invention relates to a
traffic sign recognition method based on a
capsule neural network. The method comprises the following steps: preprocessing an image by adopting methods such as image
equalization, maximum stable extremum region segmentation, normalization and the like, eliminating interference of factors such as
motion blur, background interference, illumination, local
occlusion damage of a
traffic sign and the like, and segmenting an image of a
region of interest, so that the image of the
region of interest can be effectively extracted, the recall ratio of a weak light condition isimproved, and the robustness is enhanced; in addition, a
capsule neural
network structure is introduced,
convolution layer bottom layer features are adopted, a vectorized
capsule unit is packaged after passing through a main capsule layer
tensor vector, weight parameters are updated through dynamic routing clustering and back propagation, model training and model weight parameter outputting are achieved, the training speed is high, and the
training time is shortened; and finally, image classification is realized according to the trained model weight parameters and dynamic routing clustering, so that the recall ratio of weak light pictures can be effectively improved, and the recognition rate of traffic signs is improved.