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 = []
|