hyo37009 commited on
Commit
efac9f2
·
1 Parent(s): d6467e9
Files changed (1) hide show
  1. app.py +18 -18
app.py CHANGED
@@ -18,24 +18,24 @@ model = TFSegformerForSemanticSegmentation.from_pretrained(
18
  def ade_palette():
19
  """ADE20K palette that maps each class to RGB values."""
20
  return [
21
- [255, 0, 0],
22
- [0, 255, 0],
23
- [0, 0, 255],
24
- [255, 255, 0],
25
- [0, 255, 255],
26
- [255, 0, 255],
27
- [128, 0, 0],
28
- [0, 128, 0],
29
- [0, 0, 128],
30
- [128, 128, 0],
31
- [0, 128, 128],
32
- [128, 0, 128],
33
- [192, 192, 192],
34
- [128, 128, 128],
35
- [255, 128, 0],
36
- [255, 0, 128],
37
- [0, 255, 128],
38
- [128, 255, 0]
39
  ]
40
 
41
 
 
18
  def ade_palette():
19
  """ADE20K palette that maps each class to RGB values."""
20
  return [
21
+ [255, 0, 0],
22
+ [255, 94, 0],
23
+ [255, 187, 0],
24
+ [255, 228, 0],
25
+ [171, 242, 0],
26
+ [29, 219, 22],
27
+ [0, 216, 255],
28
+ [0, 84, 255],
29
+ [1, 0, 255],
30
+ [95, 0, 255],
31
+ [255, 0, 221],
32
+ [255, 0, 127],
33
+ [152, 0, 0],
34
+ [153, 112, 0],
35
+ [107, 153, 0],
36
+ [0, 51, 153],
37
+ [63, 0, 153],
38
+ [153, 0, 133]
39
  ]
40
 
41