Real-time infrared multi-class multi-target anchor-free tracking network

被引:0
|
作者
Song Z. [1 ]
Yang J. [1 ]
Zhang D. [1 ]
Wang S. [1 ]
Zhang S. [1 ]
机构
[1] Beijing Institute of Remote Sensing Equipment, Beijing
关键词
Anchor-free; Infrared target; Multi-class; Multi-target tracking; Re-identification; Real time in edge devices;
D O I
10.12305/j.issn.1001-506X.2022.02.06
中图分类号
学科分类号
摘要
In this paper, an improved real-time infrared multi-class multi-target tracking network is proposed, which not only ensures the tracking accuracy, but also redesigns the anchor-free network structure to further reduce the number of network parameters and the network inference latency. By optimizing the target feature vector, the recognition accuracy is further improved, and the tracking process is simplified and developed. In addition, through detailed analysis of the execution time of related processes, GPU and CPU are selected to perform the optimal operation respectively to improve the overall tracking speed. The above method is applied to the low altitude sea surface infrared target tracking dataset. The results show that under the comprehensive evaluation metric proposed in this paper, the score is increased by 1.78 and the running speed of the algorithm reaches 52.37 FPS in NVIDIA Jetson Xavier NX compared with other lightweight network, which meets the real-time operation requirements of edge devices. © 2022, Editorial Office of Systems Engineering and Electronics. All right reserved.
引用
收藏
页码:401 / 409
页数:8
相关论文
共 31 条
  • [1] WANG F S, LU M Y, ZHAO Q J, Et al., Particle filtering algorithm, Chinese Journal of Computers, 37, 8, pp. 1679-1694, (2014)
  • [2] COLLINS R T., Mean-shift blob tracking through scale space, Proc. of the IEEE Computer Society Conference on Computer Vision and Pattern Recognition, (2003)
  • [3] WELCH G, BISHOP G., An introduction to the Kalman filter, Proc. of the ACM Special Interest Group on Computer Graphics and Interactive Techniques, (2001)
  • [4] BOLME D S, BEVERIDGE J R, DRAPER B A, Et al., Visual object tracking using adaptive correlation filters, Proc. of the IEEE Computer Society Conference on Computer Vision and Pattern Recognition, pp. 2544-2550, (2010)
  • [5] HENRIQUES J F, CASEIRO R, MARTINS P, Et al., Exploiting the circulant structure of tracking-by-detection with kernels, Proc. of the European Conference on Computer Vision, pp. 702-715, (2012)
  • [6] HENRIQUES J F, CASEIRO R, MARTINS P, Et al., High-speed tracking with kernelized correlation filters, IEEE Trans.on Pattern Analysis and Machine Intelligence, 37, 3, pp. 583-596, (2014)
  • [7] DU R P, ZHANG L, LU Y., Improved infrared target tracking algorithm based on context-aware correlation filter, Laser & Infrared, 50, 7, pp. 839-845, (2020)
  • [8] DANELLJAN M, ROBINSON A, SHAHBAZ K F, Et al., Beyond correlation filters: learning continuous convolution operators for visual tracking, Proc. of the European Conference on Computer Vision, pp. 472-488, (2016)
  • [9] WANG C Y, ZHANG L J, LI X M, Et al., Infrared target tracking based on improved kernel correlation filter algorithm, Electronics Optics & Control, 28, 7, pp. 9-13, (2021)
  • [10] SONG G P., Research and application of single target tracking algorithm based on siamese network, (2020)