hsshin98
commited on
Commit
·
a93577e
1
Parent(s):
a5c51bb
false save_mask
Browse files
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,))
|