Skip to content

Commit 648b7d7

Browse files
committedJun 28, 2024
fixbug: missing ignore idx for scannet200 zero-shot setting to exclude background categories
1 parent f66fbf7 commit 648b7d7

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed
 

‎tools/cfgs/scannet200_models/zs/spconv_clip_caption.yaml

+2-2
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ CLASS_NAMES: ['wall', 'chair', 'floor', 'table', 'door', 'couch', 'cabinet', 'sh
2121

2222
DATA_CONFIG:
2323
_BASE_CONFIG_: cfgs/dataset_configs/scannet200_dataset.yaml
24-
ignore_class_idx: []
24+
ignore_class_idx: [0, 2, 35, 98, 136, 147, 153, 159, 166, 167, 183, 186, 191, 198]
2525
trainonly_class_idx: [98, 136, 147, 153, 159, 166, 167, 183, 186, 191, 198]
2626

2727
CAPTION_INFO:
@@ -90,7 +90,7 @@ MODEL:
9090
TEXT_EMBED:
9191
NAME: CLIP
9292
NORM: True
93-
PATH: text_embed/scannet_clip-ViT-B16_id.pth
93+
PATH: text_embed/scannet200_clip-ViT-B16_id.pth
9494

9595
LOGIT_SCALE:
9696
value: 1.0

‎tools/cfgs/scannet200_models/zs/spconv_clip_caption_openscene.yaml

+1-1
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ CLASS_NAMES: ['wall', 'chair', 'floor', 'table', 'door', 'couch', 'cabinet', 'sh
2121

2222
DATA_CONFIG:
2323
_BASE_CONFIG_: cfgs/dataset_configs/scannet200_dataset.yaml
24-
ignore_class_idx: []
24+
ignore_class_idx: [0, 2, 35, 98, 136, 147, 153, 159, 166, 167, 183, 186, 191, 198]
2525
trainonly_class_idx: [98, 136, 147, 153, 159, 166, 167, 183, 186, 191, 198]
2626

2727
LOAD_KD_LABEL_TRAIN: True

0 commit comments

Comments
 (0)
Please sign in to comment.