timmy0079 commited on
Commit
9a037ec
·
1 Parent(s): 0cb9c32

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +35 -18
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
- [255, 255, 255],
20
- [255, 255, 0],
21
- [255, 0, 0],
22
- [0, 255, 255],
23
- [255, 0, 255],
24
- [0, 0, 255],
25
- [0, 255, 0],
26
- [255, 255, 128],
27
- [255, 128, 255],
28
- [128, 255, 255],
29
- [0, 0, 128],
30
- [0, 128, 0],
31
- [128, 0, 0],
32
- [128, 255, 128],
33
- [255, 255, 128],
34
- [128, 255, 255],
35
- [128, 0, 255],
36
- [0, 255, 128],
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 = []