hyo37009 commited on
Commit
36e05e2
·
1 Parent(s): eb36877
Files changed (1) hide show
  1. app.py +18 -155
app.py CHANGED
@@ -18,157 +18,24 @@ model = TFSegformerForSemanticSegmentation.from_pretrained(
18
  def ade_palette():
19
  """ADE20K palette that maps each class to RGB values."""
20
  return [
21
-
22
  [255, 0, 0],
23
- [0, 255, 0],
24
- [0, 0, 255],
25
- [255, 255, 0],
26
- [255, 0, 255],
27
- [0, 255, 255],
28
- [128, 0, 0],
29
- [0, 128, 0],
30
- [0, 0, 128],
31
- [128, 128, 0],
32
- [128, 0, 128],
33
- [0, 128, 128],
34
- [192, 192, 192],
35
- [128, 128, 128],
36
- [255, 128, 0],
37
- [128, 255, 0],
38
- [128, 0, 255],
39
- [45, 78, 123],
40
- [156, 200, 56],
41
- [32, 90, 210],
42
- [56, 123, 67],
43
- [180, 56, 123],
44
- [123, 67, 45],
45
- [45, 134, 200],
46
- [67, 56, 123],
47
- [78, 123, 67],
48
- [32, 210, 90],
49
- [45, 56, 189],
50
- [123, 56, 123],
51
- [56, 156, 200],
52
- [189, 56, 45],
53
- [112, 200, 56],
54
- [56, 123, 45],
55
- [200, 32, 90],
56
- [123, 45, 78],
57
- [200, 156, 56],
58
- [45, 67, 123],
59
- [56, 45, 78],
60
- [45, 56, 123],
61
- [123, 67, 56],
62
- [56, 78, 123],
63
- [210, 90, 32],
64
- [123, 56, 189],
65
- [45, 200, 134],
66
- [67, 123, 56],
67
- [123, 45, 67],
68
- [90, 32, 210],
69
- [200, 45, 78],
70
- [32, 210, 90],
71
- [45, 123, 67],
72
- [165, 42, 87],
73
- [72, 145, 167],
74
- [15, 158, 75],
75
- [209, 89, 40],
76
- [32, 21, 121],
77
- [184, 20, 100],
78
- [56, 135, 15],
79
- [128, 92, 176],
80
- [1, 119, 140],
81
- [220, 151, 43],
82
- [41, 97, 72],
83
- [148, 38, 27],
84
- [107, 86, 176],
85
- [21, 26, 136],
86
- [174, 27, 90],
87
- [91, 96, 204],
88
- [108, 50, 107],
89
- [27, 45, 136],
90
- [168, 200, 52],
91
- [7, 102, 27],
92
- [42, 93, 56],
93
- [140, 52, 112],
94
- [92, 107, 168],
95
- [17, 118, 176],
96
- [59, 50, 174],
97
- [206, 40, 143],
98
- [44, 19, 142],
99
- [23, 168, 75],
100
- [54, 57, 189],
101
- [144, 21, 15],
102
- [15, 176, 35],
103
- [107, 19, 79],
104
- [204, 52, 114],
105
- [48, 173, 83],
106
- [11, 120, 53],
107
- [206, 104, 28],
108
- [20, 31, 153],
109
- [27, 21, 93],
110
- [11, 206, 138],
111
- [112, 30, 83],
112
- [68, 91, 152],
113
- [153, 13, 43],
114
- [25, 114, 54],
115
- [92, 27, 150],
116
- [108, 42, 59],
117
- [194, 77, 5],
118
- [145, 48, 83],
119
- [7, 113, 19],
120
- [25, 92, 113],
121
- [60, 168, 79],
122
- [78, 33, 120],
123
- [89, 176, 205],
124
- [27, 200, 94],
125
- [210, 67, 23],
126
- [123, 89, 189],
127
- [225, 56, 112],
128
- [75, 156, 45],
129
- [172, 104, 200],
130
- [15, 170, 197],
131
- [240, 133, 65],
132
- [89, 156, 112],
133
- [214, 88, 57],
134
- [156, 134, 200],
135
- [78, 57, 189],
136
- [200, 78, 123],
137
- [106, 120, 210],
138
- [145, 56, 112],
139
- [89, 120, 189],
140
- [185, 206, 56],
141
- [47, 99, 28],
142
- [112, 189, 78],
143
- [200, 112, 89],
144
- [89, 145, 112],
145
- [78, 106, 189],
146
- [112, 78, 189],
147
- [156, 112, 78],
148
- [28, 210, 99],
149
- [78, 89, 189],
150
- [189, 78, 57],
151
- [112, 200, 78],
152
- [189, 47, 78],
153
- [205, 112, 57],
154
- [78, 145, 57],
155
- [200, 78, 112],
156
- [99, 89, 145],
157
- [200, 156, 78],
158
- [57, 78, 145],
159
- [78, 57, 99],
160
- [57, 78, 145],
161
- [145, 112, 78],
162
- [78, 89, 145],
163
- [210, 99, 28],
164
- [145, 78, 189],
165
- [57, 200, 136],
166
- [89, 156, 78],
167
- [145, 78, 99],
168
- [99, 28, 210],
169
- [189, 78, 47],
170
- [28, 210, 99],
171
- [78, 145, 57],
172
  ]
173
 
174
 
@@ -242,11 +109,7 @@ def sepia(input_img):
242
  demo = gr.Interface(fn=sepia,
243
  inputs=gr.Image(shape=(400, 600)),
244
  outputs=['plot'],
245
- examples=["person-1.jpg",
246
- "person-2.jpg",
247
- "person-3.jpg",
248
- "person-4.jpg",
249
- "person-5.jpg"],
250
  allow_flagging='never')
251
 
252
  demo.launch()
 
18
  def ade_palette():
19
  """ADE20K palette that maps each class to RGB values."""
20
  return [
 
21
  [255, 0, 0],
22
+ [255, 94, 0],
23
+ [255, 187, 0],
24
+ [255, 228, 0],
25
+ [171, 242, 0],
26
+ [29, 219, 22],
27
+ [0, 216, 255],
28
+ [0, 84, 255],
29
+ [1, 0, 255],
30
+ [95, 0, 255],
31
+ [255, 0, 221],
32
+ [255, 0, 127],
33
+ [152, 0, 0],
34
+ [153, 112, 0],
35
+ [107, 153, 0],
36
+ [0, 51, 153],
37
+ [63, 0, 153],
38
+ [153, 0, 133]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
39
  ]
40
 
41
 
 
109
  demo = gr.Interface(fn=sepia,
110
  inputs=gr.Image(shape=(400, 600)),
111
  outputs=['plot'],
112
+ examples=["person-1.jpg", "person-2.jpg","person-3.jpg","person-4.jpg","person-5.jpg"],
 
 
 
 
113
  allow_flagging='never')
114
 
115
  demo.launch()