gaf7921 commited on
Commit
d2d10db
·
1 Parent(s): a57f999

Update app.py

Browse files
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
+ [255, 0, 0],
21
+ [0, 255, 0],
22
+ [0, 0, 255],
23
+ [255, 255, 0],
24
+ [255, 0, 255],
25
+ [0, 255, 255],
26
+ [255, 150, 150],
27
+ [150, 255, 150],
28
+ [150, 150, 255],
29
+ [255, 255, 150],
30
+ [255, 150, 255],
31
+ [150, 255, 255],
32
+ [150, 30, 30],
33
+ [30, 150, 30],
34
+ [30, 30, 150],
35
+ [150, 150, 30],
36
+ [150, 30, 150],
37
+ [30, 150, 150]
38
  ]
39
 
40
  labels_list = []