mmpose

mmpose PyTorch=>ONNX=>TensorRT

1.Reference

• mmpose: https://github.com/open-mmlab/mmpose

2.Export ONNX Model

python3 tools/pytorch2onnx.py configs/top_down/hrnet/coco/hrnet_w48_coco_256x192.py https://download.openmmlab.com/mmpose/top_down/hrnet/hrnet_w48_coco_256x192-b9e0b3ab_20200708.pth --output-file hrnet_w48_coco_256x192.onnx

3.Build mmpose_trt Project

mkdir build && cd build
cmake ..
make -j

4.run mmpose_trt

./mmpose_trt ../config.yaml ../samples

5.detect results