test datasets

.gitignore

@@ -0,0 +1 @@
+urban*

backend.py

@@ -80,8 +80,8 @@ def visualize_image(im, mode, tree_threshold:float, building_threshold:float, co
         tree_instances = segment_tree(im, tree_threshold)
         building_instances = segment_building(im, building_threshold)
         instances = Instances.cat([tree_instances, building_instances])
-        
-    metadata = MetadataCatalog.get("predict")
+
+    metadata = MetadataCatalog.get("urban-trees-fdokv_train")
     print("metadata", type(metadata), metadata)
     print('metadata.get("thing_classes")', type(metadata.get("thing_classes")), metadata.get("thing_classes"))
 
@@ -90,10 +90,12 @@ def visualize_image(im, mode, tree_threshold:float, building_threshold:float, co
                             scale=0.5,
                             instance_mode=color_mode)
 
-    # dataset_names = MetadataCatalog.list()
-    # print(dataset_names)
+    dataset_names = MetadataCatalog.list()
+    print(dataset_names)
 
-    # category_names = metadata.get("thing_classes")
+    metadata = MetadataCatalog.get("urban-small_train")
+    category_names = metadata.get("thing_classes")
+    print(category_names)
     # visualizer = Visualizer(im[:, :, ::-1],
     #                         metadata=metadata,
     #                         scale=0.5,