ltttpku / ADA-CM

18 stars 2 forks source link

zero-shot setting #5

Closed hutuo1213 closed 8 months ago

hutuo1213 commented 8 months ago

Thank you for your excellent work. We have some questions about zero-shot. We noticed that args.num_classes is either 117 or 600. Which is used in this work? Then, can you briefly describe the code flow of zero-shot?

ltttpku commented 8 months ago

Thanks for taking an interest in our work.

We set num_classes to 117 in the zero-shot setting on HICO-DET.

For the zero-shot setting, we filter the HOI annotations related to the unseen HOI classes in the DataFactory (refer to the code between line 138 and line 159). During training, the unseen classes are ignored thanks to object_to_target (initialized here).

hutuo1213 commented 8 months ago

Thanks!