hsshin98 commited on
Commit
a93577e
1 Parent(s): a5c51bb

false save_mask

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -84,7 +84,7 @@ def predict(image, text):
84
  cfg = setup_cfg(args)
85
  demo = VisualizationDemo(cfg, text=text)
86
  predictions, visualized_output = demo.run_on_image(image)
87
- save_masks(predictions, text.split(','))
88
  canvas = fc(visualized_output.fig)
89
  canvas.draw()
90
  out = np.frombuffer(canvas.tostring_rgb(), dtype='uint8').reshape(canvas.get_width_height()[::-1] + (3,))
 
84
  cfg = setup_cfg(args)
85
  demo = VisualizationDemo(cfg, text=text)
86
  predictions, visualized_output = demo.run_on_image(image)
87
+ # save_masks(predictions, text.split(','))
88
  canvas = fc(visualized_output.fig)
89
  canvas.draw()
90
  out = np.frombuffer(canvas.tostring_rgb(), dtype='uint8').reshape(canvas.get_width_height()[::-1] + (3,))