kmc0003a commited on
Commit
c3e15b7
1 Parent(s): 13861a4

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -17,14 +17,14 @@ model = TFSegformerForSemanticSegmentation.from_pretrained(
17
  def ade_palette():
18
  """ADE20K palette that maps each class to RGB values."""
19
  return [
 
20
  [255, 0, 0],
21
  [255, 255, 0],
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],
 
17
  def ade_palette():
18
  """ADE20K palette that maps each class to RGB values."""
19
  return [
20
+ [0, 0, 0],
21
  [255, 0, 0],
22
  [255, 255, 0],
23
+ [255, 255, 255],
24
+ [255, 0, 255],
25
+ [0, 255, 0],
26
+ [0, 255, 255],
27
+ [0, 0, 255],
 
28
  [33, 147, 176],
29
  [255, 183, 76],
30
  [67, 123, 89],