Spaces:

0jung
/

clothes_segmenation

Running

김영중 commited on Nov 7, 2023

Commit

789194e

•

1 Parent(s): 06a8a47

modify segmentation

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,24 +17,24 @@ model = TFSegformerForSemanticSegmentation.from_pretrained(
 def ade_palette():
     """ADE20K palette that maps each class to RGB values."""
     return [
-        [166,166,166],
-        [255,0,0],
-        [0,216,255],
-        [134,229,127],
-        [165,102,255],
-        [229,216,92],
-        [250,237,125],
-        [196,183,59],
-        [0,0,0],
-        [0,84,255],
-        [1,0,255],
-        [255,255,255],
-        [242,150,97],
-        [204,114,61],
-        [71,200,62],
-        [47,157,39],
-        [128,65,217],
-        [255,0,221],
     ]
 labels_list = []

 def ade_palette():
     """ADE20K palette that maps each class to RGB values."""
     return [
+        [166, 166, 166],
+        [255, 0, 0],
+        [0, 216, 255],
+        [134, 229, 127],
+        [165, 102, 255],
+        [229, 216, 92],
+        [250, 237, 125],
+        [196, 183, 59],
+        [0, 0, 0],
+        [0, 84, 255],
+        [0, 51, 153],
+        [255, 255, 255],
+        [242, 150, 97],
+        [153, 56, 0],
+        [71, 200, 62],
+        [34, 116, 28],
+        [128, 65, 217],
+        [255, 0, 221],
     ]
 labels_list = []