show dataset names

backend.py

@@ -81,23 +81,23 @@ def visualize_image(im, mode="BOTH", tree_threshold=0.7, building_threshold=0.7,
         building_instances = segment_building(im, building_threshold)
         instances = Instances.cat([tree_instances, building_instances])
 
-    # visualizer = Visualizer(im[:, :, ::-1],
-    #                         scale=0.5,
-    #                         instance_mode=color_mode)
-    
-    metadata = MetadataCatalog.get("your_model_metadata")
-    category_names = metadata.get("thing_classes")
     visualizer = Visualizer(im[:, :, ::-1],
-                            metadata=metadata,
                             scale=0.5,
                             instance_mode=color_mode)
-    # in the visualizer, add category label names to detected instances
-    for instance in instances:
-        label = category_names[instance["category_id"]]
-        visualizer.draw_text(label, instance["bbox"][:2])
+
+    dataset_names = MetadataCatalog.list()
+    print(dataset_names)
+    # category_names = metadata.get("thing_classes")
+    # visualizer = Visualizer(im[:, :, ::-1],
+    #                         metadata=metadata,
+    #                         scale=0.5,
+    #                         instance_mode=color_mode)
+    # # in the visualizer, add category label names to detected instances
+    # for instance in instances:
+    #     label = category_names[instance["category_id"]]
+    #     visualizer.draw_text(label, instance["bbox"][:2])
 
     output_image = visualizer.draw_instance_predictions(instances)
     
     return Image.fromarray(output_image.get_image()[:, :, ::-1])
 
-