Spaces:
Sleeping
Sleeping
delete mouse hover, add gr.Label
Browse files
app.py
CHANGED
@@ -6,7 +6,6 @@ import numpy as np
|
|
6 |
from PIL import Image
|
7 |
import tensorflow as tf
|
8 |
from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
|
9 |
-
import mplcursors # mouse hover
|
10 |
|
11 |
feature_extractor = SegformerFeatureExtractor.from_pretrained(
|
12 |
"nielsr/segformer-b0-finetuned-segments-sidewalk",
|
@@ -117,15 +116,14 @@ def sepia(input_img):
|
|
117 |
pred_img = np.array(input_img) * 0.5 + color_seg * 0.5
|
118 |
pred_img = pred_img.astype(np.uint8)
|
119 |
|
120 |
-
fig
|
121 |
-
mplcursors.cursor(hover=True, ax=ax)
|
122 |
return fig
|
123 |
|
124 |
demo = gr.Interface(fn=sepia,
|
125 |
title="🚥Sidewalk Segmentation🏃️",
|
126 |
description="Image Segmentation for Sidewalks",
|
127 |
inputs=gr.Image(),
|
128 |
-
outputs=['plot'],
|
129 |
examples=["sidewalk_1.jpg", "sidewalk_2.jpg", "sidewalk_3.jpg", "sidewalk_4.jpg", "sidewalk_5.jpg"],
|
130 |
allow_flagging='never',
|
131 |
theme="gradio/soft",
|
|
|
6 |
from PIL import Image
|
7 |
import tensorflow as tf
|
8 |
from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
|
|
|
9 |
|
10 |
feature_extractor = SegformerFeatureExtractor.from_pretrained(
|
11 |
"nielsr/segformer-b0-finetuned-segments-sidewalk",
|
|
|
116 |
pred_img = np.array(input_img) * 0.5 + color_seg * 0.5
|
117 |
pred_img = pred_img.astype(np.uint8)
|
118 |
|
119 |
+
fig = draw_plot(pred_img, seg)
|
|
|
120 |
return fig
|
121 |
|
122 |
demo = gr.Interface(fn=sepia,
|
123 |
title="🚥Sidewalk Segmentation🏃️",
|
124 |
description="Image Segmentation for Sidewalks",
|
125 |
inputs=gr.Image(),
|
126 |
+
outputs=['plot', gr.Label(label="labels")],
|
127 |
examples=["sidewalk_1.jpg", "sidewalk_2.jpg", "sidewalk_3.jpg", "sidewalk_4.jpg", "sidewalk_5.jpg"],
|
128 |
allow_flagging='never',
|
129 |
theme="gradio/soft",
|