/HRNet_TensorRT

Primary LanguageC++Apache License 2.0Apache-2.0

HRNet_TensorRT

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