Convert rtmw-x pth 20231122 and 20231208 to ONNX

Tau-J / rtmlib

RTMPose series (RTMPose, DWPose, RTMO, RTMW) without mmcv, mmpose, mmdet etc.

Apache License 2.0

237 stars 28 forks source link

Convert rtmw-x pth 20231122 and 20231208 to ONNX #14

Open chenscottus opened 9 months ago

chenscottus commented 9 months ago

Hi,

Can you update the RTMW-X ONNX to the latest version? https://github.com/open-mmlab/mmpose/blob/main/configs/wholebody_2d_keypoint/rtmpose/cocktail14/rtmw_cocktail14.md rtmw-x_simcc-cocktail14_pt-ucoco_270e-384x288-f840f204_20231122.pth rtmw-x_simcc-cocktail14_pt-ucoco_270e-256x192-13a2546d_20231208.pth

Thanks,

-Scott

Tau-J commented 9 months ago

Hi @chenscottus , thanks for your suggest. I will update it soon. (Also, since this is an easy modification, would you like to raise a PR to fix it? Any contribution will be appreciated!)