人体姿态估计 c++版 HRNet tensorrt加速
上次用onnx模型实现了c++版的HRNet,加上NanoDet的目标检测,在我的笔记本(GTX960M)上实现了20FPS左右的帧率,参考HRNet C++实现。这次尝试使用了tensorrt对HRNet的onnx模型进行进一步加速,达到了25FPS左右的帧率。如下图所示。1.生成符合条件的onnx由于很多原因,onnx和tensorrt对一些算子的支持并不相同,有些算子在onnx上可以实现但在tensorrt上不能支持。因此需要对原来pytorch版HRNet的一些操作进行修改。HRNet的代码
原创
2021-09-23 18:21:04 ·
1942 阅读 ·
9 评论