计算机工程与应用 ›› 2021, Vol. 57 ›› Issue (4): 120-126.DOI: 10.3778/j.issn.1002-8331.1911-0351

• 模式识别与人工智能 • 上一篇    下一篇

基于PYNQ框架的深度卷积特征异构跟踪系统

崔洲涓,安军社,陈长龙,崔天舒   

  1. 1.中国科学院 国家空间科学中心 复杂航天系统电子信息技术重点实验室,北京 100190
    2.中国科学院大学,北京 100049
  • 出版日期:2021-02-15 发布日期:2021-02-06

Deep Convolutional Features Heterogeneous Tracking System Based on PYNQ Framework

CUI Zhoujuan, AN Junshe, CHEN Changlong, CUI Tianshu   

  1. 1.Key Laboratory of Electronics and Information Technology for Space Systems, National Space Science Center, Chinese Academy of Sciences, Beijing 100190, China
    2.University of Chinese Academy of Sciences, Beijing 100049, China
  • Online:2021-02-15 Published:2021-02-06

摘要:

针对深度卷积特征目标跟踪算法中特征提取计算量大、速度慢、难以在嵌入式平台上应用的问题,提出了一种基于PYNQ框架的目标跟踪方案,并将其部署在Zynq异构平台。首先设计基于深度卷积特征的目标跟踪算法;根据算法的特点进行软硬件划分,完成片上系统的构建;然后针对深度卷积特征提取的计算过程进行并行优化,导出加速IP核;最后在PYNQ框架中通过Jupyter Notebooks,使用Python语言调用加速IP核作为硬件协处理器,实现底层到顶层的数据交互。实验结果表明,算法在通用数据集OTB-2015、UAV123上取得了良好的跟踪精度;跟踪速度与未集成加速IP核时相比,提升可达30倍。在兼顾跟踪稳健性的情况下,异构跟踪系统执行效率高,可移植性好,具有工程应用价值。

关键词: PYNQ框架, 目标跟踪, 深度卷积特征, Zynq, 加速

Abstract:

In order to solve the problems of feature extraction in deep convolutional features visual tracking algorithm, such as large amount of computation, slow speed and difficulty in application on embedded platform, a visual tracking scheme based on PYNQ framework is proposed and deployed on Zynq heterogeneous platform. Firstly, a visual tracking algorithm based on deep convolutional features is designed. Then, according to the characteristics of the algorithm, the software and hardware are divided to complete the system-on-chip construction. Then, the calculation process of deep convolutional features extraction is optimized in parallel and derived as an accelerated IP core. Finally, via Jupyter Notebooks in the PYNQ framework, the accelerated IP core can be used as a hardware coprocessor to achieve data interaction from the bottom to the top. The experimental results show that the algorithm achieves good tracking accuracy on the benchmark OTB-2015 and UAV123. The tracking speed is up to 30 times higher than when the accelerated IP core is not integrated. Under the circumstance of considering the tracking robustness, the heterogeneous tracking system has high execution efficiency, good portability and engineering application value.

Key words: PYNQ framework, object tracking, deep convolutional features, Zynq, accelerate