Train with images without hand and face joints

NVlabs / imaginaire

NVIDIA's Deep Imagination Team's PyTorch Library

Other

3.99k stars 444 forks source link

Train with images without hand and face joints #105

Closed kai0122 closed 2 years ago

kai0122 commented 2 years ago

Hello,

I tried to run train.py with images and poses without hand and face data. I removed the code that extracts hand and face data, and also remove the part that visualize them. However, it keeps on telling me that I am passing "NoneType" data to the generator model.

Do you have any idea how could I solve it?

Thank you!

kai0122 commented 2 years ago

When I tried to print out the data that returned. It seems that all data["few_shot_label"], which is the ref_labels, are all None. Do you have any way to solve it?

Thank you!

tcwang0509 commented 2 years ago

Should be fixed with the latest commit now.