J Shanghai Jiaotong Univ Sci ›› 2021, Vol. 26 ›› Issue (5): 577-586.doi: 10.1007/s12204-021-2347-8

• Intelligent Connected Vehicle • Previous Articles     Next Articles

Lightweight Method for Vehicle Re-identification Using Reranking Algorithm Based on Topology Information of Surveillance Network

ZOU Yue1 (邹 悦), LI Lin2 (李 霖), YANG Xubo1 (杨旭波)   

  1. (1. School of Software, School of Electronic Information and Electrical Engineering, Shanghai Jiao Tong University, Shanghai 200240, China; 2. Shanghai International Automobile City (Group) Co., Ltd., Shanghai 201805, China)
  • Received:2020-11-30 Online:2021-10-28 Published:2021-10-28

Abstract: As an emerging visual task, vehicle re-identification refers to the identification of the same vehicle across multiple cameras. Herein, we propose a novel vehicle re-identification method that uses an improved ResNet-50 architecture and utilizes the topology information of a surveillance network to rerank the final results. In the training stage, we apply several data augmentation approaches to expand our training data and increase their diversity in a cost-effective manner. We reform the original RestNet-50 architecture by adding non-local blocks to implement the attention mechanism and replacing part of the batch normalization operations with instance batch normalization. After obtaining preliminary results from the proposed model, we use the reranking algorithm, whose core function is to improve the similarity scores of all images on the most likely path that the vehicle tends to appear to optimize the final results. Compared with most existing state-of-the-art methods, our method is lighter, requires less data annotation, and offers competitive performance.

Key words: intelligent transportation system, vehicle re-identi?cation, deep learning

CLC Number: