Journal of Jishou University(Natural Sciences Edition) ›› 2021, Vol. 42 ›› Issue (6): 15-22.DOI: 10.13438/j.cnki.jdzk.2021.06.003

• Computer • Previous Articles     Next Articles

Vehicle Re-Identification Based on Multi-Scale and Wavelet Spatial Attention

LIAO Guangkai, ZHANG Zheng, NIU Yibo, SONG Zhiguo   

  1. (1.School of Information Science and Engineering, Jishou University, Jishou 416000, Hunan China;2. School of Physics and Electromechanical Engineering, Jishou University, Jishou 416000, Hunan China)
  • Online:2021-11-25 Published:2022-01-21

Abstract: For the problem that the multi-scale vehicle re-identification model is short of extracted detail features, a re-identification model combining multi-scale vehicle features and wavelet attention mechanism is designed.  Firstly, spatial wavelet attention module is embedded into the model to obtain more details for the network. Secondly, a hierarchical fusion network is proposed, which integrates the features of different scales to improve the ability of global feature aggregation. Finally, TriHard and CrossEntropy Loss functions are used to construct the object function of vehicle identification. Experimental results of this model compared with other excellent models on VeRi and VehicleID datasets show that vehicle re-identification accuracy can be improved by embedding spatial wavelet attention into multi-scale networks.

Key words: vehicle re-identification, spatial attention mechanism, wavelet transform, multi-scale, characteristics , fusion

WeChat e-book chaoxing Mobile QQ