search

ViTAE-Transformer / ViTPose

The official repo for [NeurIPS'22] "ViTPose: Simple Vision Transformer Baselines for Human Pose Estimation" and [TPAMI'23] "ViTPose++: Vision Transformer for Generic Body Pose Estimation"
Apache License 2.0
1.38k stars 187 forks source link

Onnx version of the model #53

Closed SonNguyen2510 closed 1 year ago

SonNguyen2510 commented 2 years ago

Hi, thank you author for the great work.

I really impress with your work. By any chance could you realese the onnx version of Vitpose model? I tried to run the vitpose B* but failed many times

seaman1900 commented 1 year ago

you can try following command:

  1. download the weight and put it to the pretrained directory;
  2. i use this to export the onnx model successfully: python tools/deployment/pytorch2onnx.py configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTPose_base_coco_256x192.py pretrained/vitpose-b-multi-coco.pth