Vehicle identification method and device based on deep learning, and computer equipment

Abstract

The invention relates to the technical field of machine vision target recognition, in particular to a vehicle identification method and device based on deep learning and computer equipment. The methodcomprises the following steps: constructing a corresponding label structure tree; extracting features of the sample data by using a deep learning network; respectively calculating the probability corresponding to each node in the label structure tree; and calculating the loss of the sample data according to the probability of the node in each layer, optimizing the loss of the sample data to complete the training of the deep learning network, and classifying the sample data to identify the vehicle. According to the vehicle identification method based on deep learning, by constructing a label structure tree for a data source in advance, the internal relation between the data is utilized, and hierarchical classification is realized in the classification process according to the internal relation, so that when the deep learning neural network obtained through training optimization is used for automatic vehicle model identification, the phenomenon of vehicle brand identification errors isnot liable to occur, and the use effect of vehicle model identification is improved.

Description

technical field [0001] The present invention relates to the technical field of machine vision target recognition, in particular to a vehicle recognition method, device and computer equipment based on deep learning. Background technique [0002] Vehicle identification plays an important role in a series of application fields such as car sales, road traffic control and vehicle sales, use, after-sales maintenance and car-related products. Among them, the vehicle brand model is a very important part of the vehicle identity and plays a very important role. [0003] The existing automatic identification of vehicle brand and model can actually be understood as a classification process. That is, the extracted vehicle features are classified and classified into a category of a certain vehicle brand and model. It mainly includes two steps of vehicle localization and brand recognition in the image. Among them, vehicle positioning refers to the identification and acquisition of vehic...

AI Technical Summary

Problems solved by technology

[0007] The embodiment of the present invention provides a vehicle recognition method, device, computer equipment and storage medium based on deep learning, aiming to solve the technical problem of poor user experience caused by vehicle brand recognition errors in the prior art

Images