Added slider component

app.py

@@ -32,17 +32,21 @@ cfg.MODEL.WEIGHTS = "model_weights/model_final.pth"
 cfg.MODEL.ROI_HEADS.NUM_CLASSES = 8
 predictor = DefaultPredictor(cfg)
 
-def segment_buildings(im):
-
+def segment_buildings(im, confidence_threshold):
     im = np.array(im)
     outputs = predictor(im)
+
+    instances = outputs["instances"].to("cpu")
+    scores = instances.scores
+    selected_indices = scores > confidence_threshold
+    selected_instances = instances[selected_indices]
+
     v = Visualizer(im[:, :, ::-1],
                    scale=0.5,
                    instance_mode=ColorMode.SEGMENTATION
     )
-    print(len(outputs["instances"])," buildings detected.")
-    out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
-    
+    out = v.draw_instance_predictions(selected_instances)
+
     return Image.fromarray(out.get_image()[:, :, ::-1])
 
 # gradio components