Upload 106 files

fcclip/fcclip.py

@@ -205,17 +205,19 @@ class FCCLIP(nn.Module):
         else:
             if self.test_text_classifier is None:
                 try:
-                    nontemplated_class_names = split_labels(self.test_metadata.stuff_classes) # it includes both thing and stuff
+                    nontemplated_class_names = split_labels(self.test_metadata.stuff_classes) # it includes both thing and stuff 
                 except:
                     # this could be for insseg, where only thing_classes are available
                     nontemplated_class_names = split_labels(self.test_metadata.thing_classes)
-        
+                print("nontemplated_class_names:", nontemplated_class_names)
                 text2classifier = {}
                 test_class_names = []
                 uncached_class_name = []
                 text_classifier = []
                 # exclude those already in cache
                 for class_names in nontemplated_class_names:
+                    if not isinstance(class_names, list):
+                        class_names = [class_names]
                     for class_name in class_names:
                         if class_name in self.demo_all_text_embedding_cache:
                             text2classifier[class_name] = self.demo_all_text_embedding_cache[class_name].to(self.device)