Update app.py
Browse files
app.py
CHANGED
|
@@ -16,24 +16,41 @@ model = SegformerForSemanticSegmentation.from_pretrained(
|
|
| 16 |
def ade_palette():
|
| 17 |
"""ADE20K palette that maps each class to RGB values."""
|
| 18 |
return [
|
| 19 |
-
[
|
| 20 |
-
[
|
| 21 |
-
[
|
| 22 |
-
[
|
| 23 |
-
[
|
| 24 |
-
[
|
| 25 |
-
[
|
| 26 |
-
[
|
| 27 |
-
[
|
| 28 |
-
[
|
| 29 |
-
[
|
| 30 |
-
[
|
| 31 |
-
[
|
| 32 |
-
[
|
| 33 |
-
[
|
| 34 |
-
[
|
| 35 |
-
[
|
| 36 |
-
[
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
]
|
| 38 |
|
| 39 |
labels_list = []
|
|
|
|
| 16 |
def ade_palette():
|
| 17 |
"""ADE20K palette that maps each class to RGB values."""
|
| 18 |
return [
|
| 19 |
+
[204, 87, 92],
|
| 20 |
+
[112, 185, 212],
|
| 21 |
+
[45, 189, 106],
|
| 22 |
+
[234, 123, 67],
|
| 23 |
+
[78, 56, 123],
|
| 24 |
+
[210, 32, 89],
|
| 25 |
+
[90, 180, 56],
|
| 26 |
+
[155, 102, 200],
|
| 27 |
+
[33, 147, 176],
|
| 28 |
+
[255, 183, 76],
|
| 29 |
+
[67, 123, 89],
|
| 30 |
+
[190, 60, 45],
|
| 31 |
+
[134, 112, 200],
|
| 32 |
+
[56, 45, 189],
|
| 33 |
+
[200, 56, 123],
|
| 34 |
+
[87, 92, 204],
|
| 35 |
+
[120, 56, 123],
|
| 36 |
+
[45, 78, 123],
|
| 37 |
+
[156, 200, 56],
|
| 38 |
+
[32, 90, 210],
|
| 39 |
+
[56, 123, 67],
|
| 40 |
+
[180, 56, 123],
|
| 41 |
+
[123, 67, 45],
|
| 42 |
+
[45, 134, 200],
|
| 43 |
+
[67, 56, 123],
|
| 44 |
+
[78, 123, 67],
|
| 45 |
+
[32, 210, 90],
|
| 46 |
+
[45, 56, 189],
|
| 47 |
+
[123, 56, 123],
|
| 48 |
+
[56, 156, 200],
|
| 49 |
+
[189, 56, 45],
|
| 50 |
+
[112, 200, 56],
|
| 51 |
+
[56, 123, 45],
|
| 52 |
+
[200, 32, 90],
|
| 53 |
+
[123, 45, 78]
|
| 54 |
]
|
| 55 |
|
| 56 |
labels_list = []
|