Skip to content

Latest commit

 

History

History
24 lines (19 loc) · 592 Bytes

README.md

File metadata and controls

24 lines (19 loc) · 592 Bytes

mmpose PyTorch=>ONNX=>TensorRT

1.Reference

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