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,))
|