Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -43,7 +43,7 @@ def inference(image):
|
|
43 |
v = Visualizer(img[:, :, ::-1],
|
44 |
#metadata=val_metadata_dicts,
|
45 |
scale=0.5,
|
46 |
-
instance_mode=ColorMode.
|
47 |
)
|
48 |
# v = Visualizer(img,scale=1.2)
|
49 |
out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
|
|
|
43 |
v = Visualizer(img[:, :, ::-1],
|
44 |
#metadata=val_metadata_dicts,
|
45 |
scale=0.5,
|
46 |
+
instance_mode=ColorMode.SEGMENTATION # remove the colors of unsegmented pixels. This option is only available for segmentation models
|
47 |
)
|
48 |
# v = Visualizer(img,scale=1.2)
|
49 |
out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
|