성원 commited on
Commit
c521f12
1 Parent(s): 69cd5f6
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
- [0, 0, 0],
21
- [0, 0, 0],
22
- [0, 0, 0],
23
- [0, 0, 0],
24
- [0, 0, 0],
25
- [0, 0, 0],
26
- [0, 0, 0],
27
- [0, 0, 0],
28
- [0, 0, 0],
29
- [0, 0, 0],
30
- [0, 0, 0],
31
- [0, 0, 0],
32
- [0, 0, 0],
33
- [0, 0, 0],
34
- [0, 0, 0],
35
- [0, 0, 0],
36
- [0, 0, 0],
37
- [0, 0, 0],
38
  ]
39
 
40
  labels_list = []
 
17
  def ade_palette():
18
  """ADE20K palette that maps each class to RGB values."""
19
  return [
20
+ [156, 134, 200],
21
+ [78, 57, 189],
22
+ [200, 78, 123],
23
+ [106, 120, 210],
24
+ [145, 56, 112],
25
+ [89, 120, 189],
26
+ [185, 206, 56],
27
+ [47, 99, 28],
28
+ [112, 189, 78],
29
+ [200, 112, 89],
30
+ [89, 145, 112],
31
+ [78, 106, 189],
32
+ [112, 78, 189],
33
+ [156, 112, 78],
34
+ [28, 210, 99],
35
+ [78, 89, 189],
36
+ [189, 78, 57],
37
+ [112, 200, 78],
38
  ]
39
 
40
  labels_list = []