aerial-segmentation-model-selection

Runtime error

xwl0809 commited on Nov 6, 2023

Commit

8fc220a

•

1 Parent(s): 93804b2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,7 +38,7 @@ def segment_image(im):
     outputs = predictor(im)
     v = Visualizer(im[:, :, ::-1],
                    scale=0.5,
-                   instance_mode=ColorMode.IMAGE_BW
     )
     print(len(outputs["instances"])," trees 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"])," trees detected.")
     out = v.draw_instance_predictions(outputs["instances"].to("cpu"))