文件名称:trt_pose_model.zip
文件大小:153.59MB
文件格式:ZIP
更新时间:2023-04-08 05:04:53
trt_pose_model
https://github.com/NVIDIA-AI-IOT/trt_pose This project features multi-instance pose estimation accelerated by NVIDIA TensorRT. It is ideal for applications where low latency is necessary. It includes Training scripts to train on any keypoint task data in MSCOCO format A collection of models that may be easily optimized with TensorRT using torch2trt This project can be used easily for the task of human pose estimation, or extended for something new. Below are models pre-trained on the MSCOCO dataset. The throughput in FPS is shown for each platform
【文件预览】:
trt_pose_model
----resnet18_baseline_att_224x224_A_epoch_249.pth(81.25MB)
----densenet121_baseline_att_256x256_B_epoch_160.pth(83.52MB)