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

• • 上一篇    下一篇

  

  • 收稿日期:2020-11-30 出版日期:2021-10-28 发布日期:2021-10-28
  • 通讯作者: ZOU Yue1? (邹 悦)?E-mail: zouyue1024@163.com,YANG Xubo (杨旭波), yangxubo@sjtu.edu.cn

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.

中图分类号: