Zenn
Open2

Poseur の ONNXエクスポート の試行錯誤

PyTorch
ONNX
mmpose
PINTOPINTO
git clone https://github.com/aim-uofa/Poseur.git && cd Poseur

docker run --gpus all --rm -it \
--shm-size=10g \
-v `pwd`:/mmaction2/data \
--name mmaction2_exec_env \
pinto0309/mmaction2_exec_env:latest
PINTOPINTO
cd data

PYTHONPATH="$(dirname $0)/..":$PYTHONPATH

pip install -e . --user
pip install timm --user
  • test
python demo/top_down_img_demo.py \
configs/body/2d_kpt_sview_rgb_img/poseur/coco/poseur_res50_coco_256x192.py \
checkpoints/poseur_256x192_r50_6dec_coco.pth \
--img-root tests/data/coco/ \
--json-file tests/data/coco/test_coco.json \
--out-img-root vis_results_poseur

MODEL=poseur_mobilenetv2_coco_256x192

python3 tools/deployment/pytorch2onnx.py \
--output-file ${MODEL}.onnx \
configs/body/2d_kpt_sview_rgb_img/poseur/coco/${MODEL}.py \
checkpoints/poseur_256x192_mbv2_6dec_coco.pth