김영중 commited on
Commit
789194e
1 Parent(s): 06a8a47

modify segmentation

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
- [166,166,166],
21
- [255,0,0],
22
- [0,216,255],
23
- [134,229,127],
24
- [165,102,255],
25
- [229,216,92],
26
- [250,237,125],
27
- [196,183,59],
28
- [0,0,0],
29
- [0,84,255],
30
- [1,0,255],
31
- [255,255,255],
32
- [242,150,97],
33
- [204,114,61],
34
- [71,200,62],
35
- [47,157,39],
36
- [128,65,217],
37
- [255,0,221],
38
  ]
39
 
40
  labels_list = []
 
17
  def ade_palette():
18
  """ADE20K palette that maps each class to RGB values."""
19
  return [
20
+ [166, 166, 166],
21
+ [255, 0, 0],
22
+ [0, 216, 255],
23
+ [134, 229, 127],
24
+ [165, 102, 255],
25
+ [229, 216, 92],
26
+ [250, 237, 125],
27
+ [196, 183, 59],
28
+ [0, 0, 0],
29
+ [0, 84, 255],
30
+ [0, 51, 153],
31
+ [255, 255, 255],
32
+ [242, 150, 97],
33
+ [153, 56, 0],
34
+ [71, 200, 62],
35
+ [34, 116, 28],
36
+ [128, 65, 217],
37
+ [255, 0, 221],
38
  ]
39
 
40
  labels_list = []