search

athn-nik / teach

Official PyTorch implementation of the paper "TEACH: Temporal Action Compositions for 3D Humans"
https://teach.is.tue.mpg.de
Other
383 stars 40 forks source link

KeyError: 'test' when running python scripts/amass_splits_babel.py #4

Closed Sy-Zhang closed 1 year ago

Sy-Zhang commented 1 year ago

When running python scripts/amass_splits_babel.py, This Line https://github.com/athn-nik/teach/blob/0e49eaa2946958384d96f1539b4671b0222d93a3/scripts/amass_splits_babel.py#L47 reports KeyError: 'test'.

I found the problem is caused by dataset_db_lists, which does not have test as key. https://github.com/athn-nik/teach/blob/0e49eaa2946958384d96f1539b4671b0222d93a3/scripts/amass_splits_babel.py#L32-L33

Your paper mentioned that "we report our final results in the validation set, for easier reproduction, since BABEL test set is not publicly available". Does it mean all test sequences are discarded here?

athn-nik commented 1 year ago

You are right, I forgot to omit the test set from the process.

Sy-Zhang commented 1 year ago

Thanks for answering!