Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -35,9 +35,13 @@ def segment_buildings(im):
|
|
35 |
|
36 |
im = cv2.imread(im)
|
37 |
outputs = predictor(im)
|
38 |
-
v = Visualizer(im[:, :, ::-1],
|
|
|
|
|
|
|
39 |
out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
|
40 |
-
|
|
|
41 |
|
42 |
# gradio components
|
43 |
"""
|
|
|
35 |
|
36 |
im = cv2.imread(im)
|
37 |
outputs = predictor(im)
|
38 |
+
v = Visualizer(im[:, :, ::-1],
|
39 |
+
scale=0.5,
|
40 |
+
instance_mode=ColorMode.IMAGE_BW
|
41 |
+
)
|
42 |
out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
|
43 |
+
im = Image.fromarray(out.get_image()[:, :, ::-1])
|
44 |
+
return im
|
45 |
|
46 |
# gradio components
|
47 |
"""
|