devit
devit copied to clipboard
prototypes building
I'm trying to rebuild the prototypes in config yamls with the code provided, but it seems the keys are not the same as the .pkl and .pth files provided in Downloads.md. For example, lvis_v1_known_train.vitl14.pkl has keys(['prototypes', 'kwargs', 'label_names']) but the pkl I built with build_prototypes.ipynb only has keys(['prototypes', 'label_names']). And I also tried tools/extract_instance_prototypes.py, the pkl it built has keys(['labels', 'patch_tokens', 'avg_patch_tokens', 'image_id', 'boxes', 'areas', 'skip']) which does not match with any of the provided prototypes so I don't know where to use it.
May I know how to build prototypes exactly the same as those provided in configs yamls?