A pedestrian re-identification method based on a hole convolution and attention learning mechanism

Abstract

The invention discloses a pedestrian re-identification method based on hole convolution and an attention learning mechanism, which is used for solving the technical problem that the existing pedestrian re-identification method is poor in practicability. The technical scheme includes: firstly, desigining bottleneck modules based on hole convolution, and connecting a plurality of bottleneck modulesin series to form a trunk network; Pre-training the trunk network to obtain a pre-training model; extracting Attention feature maps at different levels of the backbone network, limiting the consistency of the attention feature maps, and learning attention features at different levels autonomously; Training the network by adopting a cross entropy loss function, a triple loss function and an attention feature map constraint loss function; And directly extracting a final feature by utilizing the trunk network, searching a pedestrian image with the minimum distance from the feature of the to-be-searched pedestrian in a pedestrian retrieval library, endowing the identity to the to-be-searched pedestrian, and finishing a re-identification process. According to the method, the convolutional neural network and the attention learning mechanism are combined, pedestrian re-identification can be accurately carried out, and the practicability is good.

Description

technical field [0001] The invention relates to a pedestrian re-identification method, in particular to a pedestrian re-identification method based on hole convolution and attention learning mechanism. Background technique [0002] Pedestrian re-identification refers to the technology of identifying the identity of pedestrians in different camera scenarios, which is a very important part of video surveillance analysis technology. However, due to the complexity of surveillance video, the impact of drastic changes in lighting, weather, viewing angle changes, pedestrian posture, occlusion and other factors, as well as the poor resolution of imaging equipment, it is difficult to identify the same pedestrian under different cameras. With deep learning, major breakthroughs have been made in many computer vision fields such as image classification and object recognition. [0003] Applying deep learning to pedestrian re-identification can deal with the above problems well. General...

AI Technical Summary

Problems solved by technology

[0011] In order to overcome the shortcomings of poor practicability of existing pedestrian re-identification methods, the present invention provides a pedestrian re-identification method based on hole convolution and attention learning mechanism

Images