ccvs icon indicating copy to clipboard operation
ccvs copied to clipboard

example script for conditioning on category

Open nmochar2 opened this issue 2 years ago • 1 comments

Hi,

Do you have an example bash script for how to train CCVS with class-conditioning? I'm trying to replicate the Weizmann results.

Thank you

nmochar2 avatar Oct 31 '22 20:10 nmochar2

Hi,

Thanks for your interest.

You should be able to do this just by using the flag --x_cat when training the transformer, and the flag --categories which enumerates the classes that interest you (e.g., jumping walking, running ...), and by having your BaseDataset class return the index corresponding to the classes in ground-truth videos for training.

Hope this helps!

16lemoing avatar Nov 05 '22 16:11 16lemoing