Vehicle fine-grained classification method and device

Abstract

The invention provides a vehicle fine-grained classification method and device. The vehicle fine-grained classification method comprises the steps of extracting a basic feature map of an input picturethrough a convolutional neural network; adaptively constructing a global structure diagram according to the basic feature diagram; adopting a graph convolutional neural network to carry out association relationship reasoning on the global structure graph to form global guidance information; applying the global guidance information to the basic feature map to obtain an enhanced feature map; and inputting the enhanced feature map into a classifier for classification. According to the vehicle fine-grained image classification method, association relationship reasoning is carried out on differentparts of the vehicle, and the association relationship reasoning result is applied to the basic feature map, so that the obtained feature map has higher expression capability, and the vehicle fine-grained image classification accuracy is improved; in addition, the classification method does not need to mark key areas, and is quick in feature extraction, wide in application scene and low in calculation consumption.

Description

technical field [0001] The invention relates to the technical field of vehicle classification, in particular to a fine-grained vehicle classification method and device. Background technique [0002] As an important means of transportation for human travel, vehicles are closely related to people's lives, and the development of modern vehicle production technology has also made more and more types of vehicles appear in people's daily life; vehicles usually have similar appearance structures, Different types of vehicles only have slight differences in the positions of lights, doors, etc., and fine-grained identification of vehicles often requires experts in related fields. [0003] The fine-grained classification of vehicles is one of the most important tasks in computer vision. It not only plays an important role in other vision tasks, such as vehicle re-identification, vehicle tracking, but also provides assistance for the city's brain and intelligent traffic construction; at...

AI Technical Summary

Problems solved by technology

[0013] In view of this, the purpose of the present invention is to propose a vehicle fine-grained classification method to solve the problems existing in the existing classification methods that the key area information needs to be marked, the key areas are not related, and the calculation consumption is large

Images