HRNet的TensorRT加速,编译器是VS2019,实现过程讲解可以看博客人体姿态估计 c++版 HRNet tensorrt加速
关于环境的配置可以自行百度。主要是opencv的cuda版、onnx的gpu推理库以及TensorRT Windows版本的配置。该项目是Release x64版本
打开sln文件选择Release x64版本,会自带我电脑的环境配置,但编译需要自己重新配,因为那是我电脑的环境。
最后的程序会将onnx模型转化为trt,trt文件和gpu相关,所以只上传onnx模型,可能会缺dll,百度自行解决。
Release程序上传至百度网盘,链接:HRNet_TensorRT_Release 提取码:ucef
使用方法:
HRNet-C++.exe -v E:\ytxs111.mp4 -m 2
参数的含义如下,-h可以查看
Usage: example [options...]
Options:
-v, --video Video path
-c, --camera camera index
-m, --model model type,0-w48_256x192,1-w48_384x288,2-w32_256x192,3-w32_128x96
-d, --display point display mode, 0-左右,1-左,2-右
-w, --write_video write video path
-h, --help Shows this page