search

BrianG13 / FLEX

FLEX: Parameter-free Multi-view 3D Human Motion Reconstruction
47 stars 3 forks source link

Training on custom dataset #20

Open UsamaShaikh1 opened 1 year ago

UsamaShaikh1 commented 1 year ago

Hi, I am trying the code base with a custom dataset with a different number of keypoints. But the model for some reason not learning angles and even after 300 epochs the angle prediction is not even near gt angles. Can you please guide me on this issue. @BrianG13

BrianG13 commented 1 year ago

Hi @UsamaShaikh1 , can you add here your losses plot?

What data are you using to train the model? Please note that if you changed the number of keypoints, you need to change the S branch as well to output the correct number of bones and change the FK layer with your body topology