Spaces:
Sleeping
Sleeping
Commit
·
6442ed7
1
Parent(s):
cc37cda
Update app.py
Browse files
app.py
CHANGED
@@ -100,7 +100,8 @@ def sepia(input_img):
|
|
100 |
for label in unique_labels:
|
101 |
mask = (seg.numpy() == label)
|
102 |
class_name = labels_list[label]
|
103 |
-
class_prob =
|
|
|
104 |
class_probabilities[class_name] = class_prob
|
105 |
|
106 |
# Gradio Interface에 출력할 문자열 생성
|
@@ -123,12 +124,3 @@ demo = gr.Interface(fn=sepia,
|
|
123 |
|
124 |
demo.launch()
|
125 |
|
126 |
-
|
127 |
-
|
128 |
-
demo = gr.Interface(fn=sepia,
|
129 |
-
inputs=gr.Image(shape=(400, 600)),
|
130 |
-
outputs=['plot', 'text'],
|
131 |
-
examples=["citiscapes-1.jpeg", "citiscapes-2.jpeg"],
|
132 |
-
allow_flagging='never')
|
133 |
-
|
134 |
-
demo.launch()
|
|
|
100 |
for label in unique_labels:
|
101 |
mask = (seg.numpy() == label)
|
102 |
class_name = labels_list[label]
|
103 |
+
class_prob = tf.nn.softmax(logits.numpy()[0][:, :, label]) # softmax 적용
|
104 |
+
class_prob = np.mean(class_prob[mask])
|
105 |
class_probabilities[class_name] = class_prob
|
106 |
|
107 |
# Gradio Interface에 출력할 문자열 생성
|
|
|
124 |
|
125 |
demo.launch()
|
126 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|