search

muelea / buddi

Our model BUDDI learns the joint distribution of interacting people
https://muelea.github.io/buddi/
MIT License
140 stars 12 forks source link

Data splits for all datasets #9

Open laubravo opened 8 months ago

laubravo commented 8 months ago

Hi,

While you make the 'train_val_split.npz' files available, can you specify what the data splits are for all the datasets (Flickr, CHI3D and Hi4D)?

Thank you!

muelea commented 8 months ago

Hi laubravo,

you can download the train/val/test splits from here

The Flickr Fits data for val/test should be available later today.

Best wishes, Lea

laubravo commented 8 months ago

can you provide more details on what the 'images_contact_projected_joints_2d.pkl' file used in the preprocessing correspondence script should be or how it's generated?

muelea commented 8 months ago

These are the GT 3D joints from Hi4D and CHI3D projected onto the image plane to get the people in contact and then solve the correspondence with BEV and OpenPose/ViTPose keypoints. I pushed the two scripts: CHI3D Hi4D