devit icon indicating copy to clipboard operation
devit copied to clipboard

Why are the shapes of these two files different? ycb_prototypes.pth VS OVD prototypes

Open Anymake opened this issue 1 year ago • 1 comments

Why are the shapes of these two files different? ycb_prototypes.pth :torch.Size([34, 1024]) weights/initial/open-vocabulary/prototypes/lvis/lvis_v1_novel_train.vitl14.pkl :torch.Size([337, 10, 1024]) weights/initial/few-shot/prototypes/fs_coco_trainval_novel_5shot.vitl14.pkl :torch.Size([20, 10, 1024]) I know dim0 is class num, but why the prototype of one class in demo is 1x1024, but in the eval is 10 x 1024 ?

Anymake avatar Nov 30 '23 03:11 Anymake