Spaces:
Sleeping
Sleeping
Karin0616
commited on
Commit
·
683e020
1
Parent(s):
f5feee4
test
Browse files
app.py
CHANGED
@@ -17,24 +17,24 @@ model = TFSegformerForSemanticSegmentation.from_pretrained(
|
|
17 |
def ade_palette():
|
18 |
"""ADE20K palette that maps each class to RGB values."""
|
19 |
return [
|
20 |
-
[204, 87, 92],
|
21 |
-
[112, 185, 212],
|
22 |
-
[
|
23 |
-
[
|
24 |
-
[
|
25 |
-
[
|
26 |
-
[
|
27 |
-
[
|
28 |
-
[
|
29 |
-
[255,
|
30 |
-
[
|
31 |
-
[
|
32 |
-
[
|
33 |
-
[
|
34 |
-
[
|
35 |
-
[
|
36 |
-
[
|
37 |
-
[
|
38 |
]
|
39 |
|
40 |
labels_list = []
|
|
|
17 |
def ade_palette():
|
18 |
"""ADE20K palette that maps each class to RGB values."""
|
19 |
return [
|
20 |
+
[204, 87, 92], # background (Reddish)
|
21 |
+
[112, 185, 212], # hat (Blue)
|
22 |
+
[196, 160, 122], # hair (Brown)
|
23 |
+
[106, 135, 242], # sunglasses (Light Blue)
|
24 |
+
[91, 192, 222], # upper-clothes (Turquoise)
|
25 |
+
[255, 192, 203], # skirt (Pink)
|
26 |
+
[176, 224, 230], # pants (Light Blue)
|
27 |
+
[222, 49, 99], # dress (Red)
|
28 |
+
[139, 69, 19], # belt (Brown)
|
29 |
+
[255, 0, 0], # left-shoe (Red)
|
30 |
+
[0, 0, 255], # right-shoe (Blue)
|
31 |
+
[255, 228, 181], # face (Peach)
|
32 |
+
[128, 0, 0], # left-leg (Maroon)
|
33 |
+
[0, 128, 0], # right-leg (Green)
|
34 |
+
[255, 99, 71], # left-arm (Tomato)
|
35 |
+
[0, 255, 0], # right-arm (Lime)
|
36 |
+
[128, 0, 128], # bag (Purple)
|
37 |
+
[255, 255, 0] # scarf (Yellow)
|
38 |
]
|
39 |
|
40 |
labels_list = []
|