search

Graformer / GraFormer

The Pytorch implementation for "GraFormer: Graph Convolution Transformer for 3D Pose Estimation" https://arxiv.org/pdf/2109.08364.pdf
Apache License 2.0
51 stars 6 forks source link

Cannot use the CPN data to train and test the model #4

Open GoogQing opened 2 years ago

GoogQing commented 2 years ago

I downloaded the 2D CPN data from VideoPose3D but failed to use them to train and test the model. Could you help to share the preprocessed 2D CPN data?

rongtian1996 commented 1 year ago

Hello, when i train the model, i always get the error "RuntimeError: Expected size for first two dimensions of batch2 tensor to be: [192, 16] but got: [192, 17].", and i can't solve it. Did you meet the problem,? Can you help me? Thankyou very much.

GoogQing commented 1 year ago

Hi, you need to remove the nose joint. This project does not predict this joint. :)

rongtian1996 commented 1 year ago

In the project h36m_dataset.py, line 271 , the code has removed the nose joint . Where do i need to remove the nose joint? In the process of dataset setup? Can you be more specific? Thankyou.

dyf111 commented 2 months ago

In the project h36m_dataset.py, line 271 , the code has removed the nose joint . Where do i need to remove the nose joint? In the process of dataset setup? Can you be more specific? Thankyou.

did you resolve this problem?how ?