用TensorRT的人体姿势预测

Github
239 0 2020-01-23

本项目采用NVIDIA TensorRT加速多实例位姿估计。它非常适合需要低延迟的应用程序。它包括:

使用MSCOCO格式对任何关键点任务数据进行训练的训练脚本

可以使用torch2trt使用TensorRT轻松优化的模型集合


该项目可以方便地用于人体姿态估计任务,也可以扩展到一些新的领域。

具体介绍看这里:

https://github.com/NVIDIA-AI-IOT/trt_pose