Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -32,7 +32,7 @@ cfg.MODEL.ROI_HEADS.NUM_CLASSES = 8
|
|
32 |
|
33 |
def segment_buildings(input_image):
|
34 |
|
35 |
-
im = cv2.imread(input_image
|
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"))
|
|
|
32 |
|
33 |
def segment_buildings(input_image):
|
34 |
|
35 |
+
im = cv2.imread(input_image)
|
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"))
|