search

mks0601 / 3DMPPE_ROOTNET_RELEASE

Official PyTorch implementation of "Camera Distance-aware Top-down Approach for 3D Multi-person Pose Estimation from a Single RGB Image", ICCV 2019
MIT License
477 stars 65 forks source link

Converting to onnx #52

Open vigneshrk29 opened 1 year ago

vigneshrk29 commented 1 year ago

Hi,

I am trying to convert the model to onnx format and would like to know what the input shape should be? The model takes in 2 inputs doesnt it

Thanks

mks0601 commented 1 year ago

(256,256) shape of image and a scalar of gamma.

vigneshrk29 commented 1 year ago

Hi,

Thanks. Sorry but if I had to put it into a format like : [batch_size, channels, height, width],

how would I go about doing this?