Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -34,9 +34,10 @@ predictor = DefaultPredictor(cfg)
|
|
34 |
def segment_buildings(im):
|
35 |
|
36 |
outputs = predictor(im)
|
37 |
-
v = Visualizer(im[:, :, ::-1], scale=1.2)
|
38 |
out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
|
39 |
-
|
|
|
40 |
|
41 |
# gradio components
|
42 |
"""
|
|
|
34 |
def segment_buildings(im):
|
35 |
|
36 |
outputs = predictor(im)
|
37 |
+
v = Visualizer(im[:, :, ::-1], MetadataCatalog.get(cfg.DATASETS.TRAIN[0]), scale=1.2)
|
38 |
out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
|
39 |
+
out_im = out.get_image()[:, :, ::-1]
|
40 |
+
return out_im
|
41 |
|
42 |
# gradio components
|
43 |
"""
|