yucornetto commited on
Commit
2df50c2
1 Parent(s): 12fc175

Upload 106 files

Browse files
Files changed (2) hide show
  1. fcclip/.DS_Store +0 -0
  2. fcclip/fcclip.py +4 -2
fcclip/.DS_Store CHANGED
Binary files a/fcclip/.DS_Store and b/fcclip/.DS_Store differ
 
fcclip/fcclip.py CHANGED
@@ -205,17 +205,19 @@ class FCCLIP(nn.Module):
205
  else:
206
  if self.test_text_classifier is None:
207
  try:
208
- nontemplated_class_names = split_labels(self.test_metadata.stuff_classes) # it includes both thing and stuff
209
  except:
210
  # this could be for insseg, where only thing_classes are available
211
  nontemplated_class_names = split_labels(self.test_metadata.thing_classes)
212
-
213
  text2classifier = {}
214
  test_class_names = []
215
  uncached_class_name = []
216
  text_classifier = []
217
  # exclude those already in cache
218
  for class_names in nontemplated_class_names:
 
 
219
  for class_name in class_names:
220
  if class_name in self.demo_all_text_embedding_cache:
221
  text2classifier[class_name] = self.demo_all_text_embedding_cache[class_name].to(self.device)
 
205
  else:
206
  if self.test_text_classifier is None:
207
  try:
208
+ nontemplated_class_names = split_labels(self.test_metadata.stuff_classes) # it includes both thing and stuff
209
  except:
210
  # this could be for insseg, where only thing_classes are available
211
  nontemplated_class_names = split_labels(self.test_metadata.thing_classes)
212
+ print("nontemplated_class_names:", nontemplated_class_names)
213
  text2classifier = {}
214
  test_class_names = []
215
  uncached_class_name = []
216
  text_classifier = []
217
  # exclude those already in cache
218
  for class_names in nontemplated_class_names:
219
+ if not isinstance(class_names, list):
220
+ class_names = [class_names]
221
  for class_name in class_names:
222
  if class_name in self.demo_all_text_embedding_cache:
223
  text2classifier[class_name] = self.demo_all_text_embedding_cache[class_name].to(self.device)