Multi-Object Tracking in Adaptive Region (tentative)
该项目实现了一个面向物联网端边环境下的多目标跟踪应用并行加速系统,它于内部集成了我所设计开发的三种算法加速方案,并给出了一套带有用户交互界面的客户端程序以测试与查看加速方案的执行效果。
除了本README外,可以阅读项目中的 paper 获取更多信息。
- (2022.6.29):3rd_place_medal:本项目工作被江苏省教育厅评选为江苏省优秀毕业论文(设计),获三等奖!🎉🎉🎉
- (2022.3.27):mega:本项目的工作在CBD2021会议上进行了线上报告。slides
本项目涉及边缘和终端两台设备,另外你可能还需要一台路由器以提供Wifi连接服务。
- GPU:NVIDIA GTX1050
- CPU:Intel i5-7300HQ
- 内存:8G
NVIDIA Jetson Nano设备
- GPU:NVIDIA Maxwell 128Core
- CPU:ARM A57 4Core
- 内存:2G
- Ubuntu 20.04
- CUDA 11.2
- TensorRT 7.2.3
- Python 3.8.5
- PyTorch 1.8.1
- GCC 9.3.0
- JetPack 4.5
- CUDA 10.2
- Python 3.6.9
- PyTorch 1.8.0
- 基于FairMOT算法
- 针对多目标跟踪算法特性所设计出了一种时空间执行优化策略
- 在边缘设备上实现了基于TensorRT的模型加速
- 设计使用了一种自适应端边算力和网络动态环境的端边协同机制
- 使用时空间执行优化策略,能够在原始处理仅能达到3.05fps的机器上也能获得24.05fps,基本达到实时分析需求,代价是损失12.8%绝对跟踪精度
- TensorRT引擎能达到1.53的预测模型推理加速比以及1.16的应用处理加速比
- 自适应端边协同机制的使用相对终端本地执行能达到4.42的加速比,同时具备网络环境适应性
- 结合使用三种加速方案,通过分别损失3%、7.1%和10.4%的MOTA准确度,能够在本地执行仅能达到0.26FPS的Jetson Nano 2G设备上达到4.58、7.7和9.91FPS的处理吞吐率。
-
安装好CUDA、CuDNN等深度学习底层支持组件
-
安装一些来自第三方的软件库
sh install.sh
-
安装一些必要的Python包
pip install -r requirements.txt
-
下载原版权重 这里给出原版仓库所给出的一些链接 fairmot_dla34.pth [Google] [Baidu, 提取码:uouv] [Onedrive]。下载完成后将权重文件放到weights文件夹下。
mv fairmot_dla34.pth ./weights/
-
在Jetson Nano上安装好JetPack4.5 OS包 JetPack4.5 Archive
-
安装一些来自第三方的软件库
sh install.sh
-
安装一些必要的Python包
pip install -r requirements.txt
-
下载原版权重 这里给出原版仓库所给出的一些链接 fairmot_dla34.pth Google Baidu, 提取码:uouv Onedrive。下载完成后将权重文件放到weights文件夹下。
mv fairmot_dla34.pth ./weights/
-
编译Plugin
cd Software/trt_impl/ make -j
-
导出PyTorch模型为ONNX模型
python build_onnx_fairmot.py python replace_dcn_plugin.py
-
将ONNX模型转化为TensorRT Engine
sh build_trt_dyn.sh
-
运行程序
cd ../MOTsc python main.py
-
跟随演示视频进行交互操作
终端上的使用需要配合边缘客户端程序进行,具体交互操作详见演示视频
-
进入终端程序文件夹
cd Software/MOTsc/end
-
启动终端client
python client.py
基础优化对象FairMOT
- 本项目中的PyTorch版本FairMOT代码大部分来自于它的原版作者ifzhang的FairMOT
- 由于在 PyTorch1.8.1+cu11 版本中不能成功运行来自FairMOT原版仓库的DCNv2算子了,所以使用了一份来自tteepe的JIT版本DCNv2算子DCNv2代替它
TensorRT工作
- 本项目中的TensorRT加速相关工作也可详见本人的竞赛项目trt-fairmot
If this work helps your research or work, please consider citing the following works:
@INPROCEEDINGS{MOTinAR,
author={Liu, Mengyang and Tang, Anran and Wang, Huitian and Shen, Lin and Chang, Yunhan and Cai, Guangxing and Yin, Daheng and Dong, Fang and Zhao, Wei},
booktitle={2021 Ninth International Conference on Advanced Cloud and Big Data (CBD)},
title={Accelerating Multi-Object Tracking in Edge Computing Environment with Time-Spatial Optimization},
year={2022},
volume={},
number={},
pages={279-284},
doi={10.1109/CBD54617.2021.00055}
}