Hugging Face's logo

Spaces:
taewon99
/
cloth-segmentation
Runtime error

App Files Community
taewon99 commited on
Commit
de9af89
·
1 Parent(s): bb9629e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -18
app.py CHANGED
@@ -18,24 +18,24 @@ model = TFSegformerForSemanticSegmentation.from_pretrained(
18
  def ade_palette():
19
  """ADE20K palette that maps each class to RGB values."""
20
  return [
21
- [230, 183, 173],
22
- [232, 181, 201],
23
- [234, 208, 179],
24
- [217, 192, 198],
25
- [215, 195, 203],
26
- [206, 166, 191],
27
- [215, 162, 205],
28
- [205, 176, 194],
29
- [234, 196, 184],
30
- [210, 181, 186],
31
- [220, 152, 174],
32
- [219, 187, 206],
33
- [224, 172, 181],
34
- [220, 162, 203],
35
- [219, 188, 160],
36
- [208, 178, 169],
37
- [222, 196, 203],
38
- [208, 153, 190],
39
  ]
40
 
41
 
 
18
  def ade_palette():
19
  """ADE20K palette that maps each class to RGB values."""
20
  return [
21
+ [255, 0, 0],
22
+ [0, 255, 0],
23
+ [0, 0, 255],
24
+ [255, 255, 0],
25
+ [0, 255, 255],
26
+ [255, 0, 255],
27
+ [128, 0, 0],
28
+ [0, 128, 0],
29
+ [0, 0, 128],
30
+ [128, 128, 0],
31
+ [0, 128, 128],
32
+ [128, 0, 128],
33
+ [192, 192, 192],
34
+ [128, 128, 128],
35
+ [255, 128, 0],
36
+ [255, 0, 128],
37
+ [0, 255, 128],
38
+ [128, 255, 0]
39
  ]
40
 
41