Spaces:

SIH
/

building-segmentation

Runtime error

xwl0809 commited on Nov 6, 2023

Commit

84b4836

•

1 Parent(s): 9810614

patch: change colour mode

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,7 +38,7 @@ def segment_buildings(im):
     outputs = predictor(im)
     v = Visualizer(im[:, :, ::-1],
                    scale=0.5,
-                   instance_mode=ColorMode.IMAGE_BW
     )
     print(len(outputs["instances"])," buildings detected.")
     out = v.draw_instance_predictions(outputs["instances"].to("cpu"))

     outputs = predictor(im)
     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"))