Karin0616 commited on
Commit
683e020
·
1 Parent(s): f5feee4
Files changed (1) hide show
  1. app.py +18 -18
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
- [45, 189, 106],
23
- [234, 123, 67],
24
- [78, 56, 123],
25
- [210, 32, 89],
26
- [90, 180, 56],
27
- [155, 102, 200],
28
- [33, 147, 176],
29
- [255, 183, 76],
30
- [67, 123, 89],
31
- [190, 60, 45],
32
- [134, 112, 200],
33
- [56, 45, 189],
34
- [200, 56, 123],
35
- [87, 92, 204],
36
- [120, 56, 123],
37
- [45, 78, 123]
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 = []