ruiite commited on
Commit
fc9264e
1 Parent(s): 84cedfd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.IMAGE_BW # 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"))
 
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"))