Spaces:
Runtime error
Runtime error
modify
Browse files- app.py +3 -3
- labels.txt +52 -52
app.py
CHANGED
@@ -10,10 +10,10 @@ import requests
|
|
10 |
|
11 |
|
12 |
feature_extractor = MaskFormerImageProcessor.from_pretrained(
|
13 |
-
"
|
14 |
)
|
15 |
model = MaskFormerForInstanceSegmentation.from_pretrained(
|
16 |
-
"
|
17 |
)
|
18 |
|
19 |
def ade_palette():
|
@@ -168,7 +168,7 @@ def ade_palette():
|
|
168 |
[99, 28, 210],
|
169 |
[189, 78, 47],
|
170 |
[28, 210, 99],
|
171 |
-
[141, 21, 43]
|
172 |
]
|
173 |
|
174 |
labels_list = []
|
|
|
10 |
|
11 |
|
12 |
feature_extractor = MaskFormerImageProcessor.from_pretrained(
|
13 |
+
"nvidia/segformer-b5-finetuned-cityscapes-1024-1024"
|
14 |
)
|
15 |
model = MaskFormerForInstanceSegmentation.from_pretrained(
|
16 |
+
"nvidia/segformer-b5-finetuned-cityscapes-1024-1024"
|
17 |
)
|
18 |
|
19 |
def ade_palette():
|
|
|
168 |
[99, 28, 210],
|
169 |
[189, 78, 47],
|
170 |
[28, 210, 99],
|
171 |
+
[141, 21, 43],
|
172 |
]
|
173 |
|
174 |
labels_list = []
|
labels.txt
CHANGED
@@ -4,23 +4,23 @@ sky
|
|
4 |
floor
|
5 |
tree
|
6 |
ceiling
|
7 |
-
road
|
8 |
bed
|
9 |
-
|
10 |
grass
|
11 |
cabinet
|
12 |
-
sidewalk
|
13 |
person
|
14 |
-
earth
|
15 |
door
|
16 |
table
|
17 |
-
mountain
|
18 |
plant
|
19 |
curtain
|
20 |
chair
|
21 |
car
|
22 |
water
|
23 |
-
painting
|
24 |
sofa
|
25 |
shelf
|
26 |
house
|
@@ -32,95 +32,95 @@ armchair
|
|
32 |
seat
|
33 |
fence
|
34 |
desk
|
35 |
-
rock
|
36 |
-
wardrobe
|
37 |
lamp
|
38 |
-
|
39 |
-
|
40 |
cushion
|
41 |
-
base
|
42 |
box
|
43 |
-
column
|
44 |
-
signboard
|
45 |
-
chest of drawers
|
46 |
counter
|
47 |
sand
|
48 |
sink
|
49 |
skyscraper
|
50 |
fireplace
|
51 |
-
refrigerator
|
52 |
-
grandstand
|
53 |
path
|
54 |
stairs
|
55 |
runway
|
56 |
-
case
|
57 |
-
pool table
|
58 |
pillow
|
59 |
-
screen door
|
60 |
-
stairway
|
61 |
river
|
62 |
-
bridge
|
63 |
bookcase
|
64 |
-
blind
|
65 |
coffee table
|
66 |
-
toilet
|
67 |
flower
|
68 |
book
|
69 |
hill
|
70 |
bench
|
71 |
countertop
|
72 |
stove
|
73 |
-
palm
|
74 |
kitchen island
|
75 |
computer
|
76 |
swivel chair
|
77 |
boat
|
78 |
bar
|
79 |
arcade machine
|
80 |
-
hovel
|
81 |
bus
|
82 |
towel
|
83 |
light
|
84 |
truck
|
85 |
tower
|
86 |
chandelier
|
87 |
-
awning
|
88 |
-
|
89 |
booth
|
90 |
-
|
91 |
-
|
92 |
dirt track
|
93 |
-
|
94 |
pole
|
95 |
-
land
|
96 |
-
bannister
|
97 |
-
escalator
|
98 |
-
ottoman
|
99 |
bottle
|
100 |
-
buffet
|
101 |
-
poster
|
102 |
stage
|
103 |
van
|
104 |
ship
|
105 |
fountain
|
106 |
-
conveyer belt
|
107 |
canopy
|
108 |
-
washer
|
109 |
-
plaything
|
110 |
-
pool
|
111 |
stool
|
112 |
-
barrel
|
113 |
-
basket
|
114 |
-
|
115 |
tent
|
116 |
bag
|
117 |
-
minibike
|
118 |
cradle
|
119 |
oven
|
120 |
ball
|
121 |
-
food
|
122 |
-
step
|
123 |
-
tank
|
124 |
trade name
|
125 |
microwave
|
126 |
pot
|
@@ -129,22 +129,22 @@ bicycle
|
|
129 |
lake
|
130 |
dishwasher
|
131 |
screen
|
132 |
-
blanket
|
133 |
sculpture
|
134 |
-
hood
|
135 |
sconce
|
136 |
vase
|
137 |
traffic light
|
138 |
tray
|
139 |
-
|
140 |
fan
|
141 |
pier
|
142 |
-
|
143 |
plate
|
144 |
monitor
|
145 |
bulletin board
|
146 |
shower
|
147 |
radiator
|
148 |
-
glass
|
149 |
clock
|
150 |
flag
|
|
|
4 |
floor
|
5 |
tree
|
6 |
ceiling
|
7 |
+
road
|
8 |
bed
|
9 |
+
windowpane
|
10 |
grass
|
11 |
cabinet
|
12 |
+
sidewalk
|
13 |
person
|
14 |
+
earth
|
15 |
door
|
16 |
table
|
17 |
+
mountain
|
18 |
plant
|
19 |
curtain
|
20 |
chair
|
21 |
car
|
22 |
water
|
23 |
+
painting
|
24 |
sofa
|
25 |
shelf
|
26 |
house
|
|
|
32 |
seat
|
33 |
fence
|
34 |
desk
|
35 |
+
rock
|
36 |
+
wardrobe
|
37 |
lamp
|
38 |
+
bathtub
|
39 |
+
railing
|
40 |
cushion
|
41 |
+
base
|
42 |
box
|
43 |
+
column
|
44 |
+
signboard
|
45 |
+
chest of drawers
|
46 |
counter
|
47 |
sand
|
48 |
sink
|
49 |
skyscraper
|
50 |
fireplace
|
51 |
+
refrigerator
|
52 |
+
grandstand
|
53 |
path
|
54 |
stairs
|
55 |
runway
|
56 |
+
case
|
57 |
+
pool table
|
58 |
pillow
|
59 |
+
screen door
|
60 |
+
stairway
|
61 |
river
|
62 |
+
bridge
|
63 |
bookcase
|
64 |
+
blind
|
65 |
coffee table
|
66 |
+
toilet
|
67 |
flower
|
68 |
book
|
69 |
hill
|
70 |
bench
|
71 |
countertop
|
72 |
stove
|
73 |
+
palm
|
74 |
kitchen island
|
75 |
computer
|
76 |
swivel chair
|
77 |
boat
|
78 |
bar
|
79 |
arcade machine
|
80 |
+
hovel
|
81 |
bus
|
82 |
towel
|
83 |
light
|
84 |
truck
|
85 |
tower
|
86 |
chandelier
|
87 |
+
awning
|
88 |
+
streetlight
|
89 |
booth
|
90 |
+
television receiver
|
91 |
+
airplane
|
92 |
dirt track
|
93 |
+
apparel
|
94 |
pole
|
95 |
+
land
|
96 |
+
bannister
|
97 |
+
escalator
|
98 |
+
ottoman
|
99 |
bottle
|
100 |
+
buffet
|
101 |
+
poster
|
102 |
stage
|
103 |
van
|
104 |
ship
|
105 |
fountain
|
106 |
+
conveyer belt
|
107 |
canopy
|
108 |
+
washer
|
109 |
+
plaything
|
110 |
+
swimming pool
|
111 |
stool
|
112 |
+
barrel
|
113 |
+
basket
|
114 |
+
waterfall
|
115 |
tent
|
116 |
bag
|
117 |
+
minibike
|
118 |
cradle
|
119 |
oven
|
120 |
ball
|
121 |
+
food
|
122 |
+
step
|
123 |
+
tank
|
124 |
trade name
|
125 |
microwave
|
126 |
pot
|
|
|
129 |
lake
|
130 |
dishwasher
|
131 |
screen
|
132 |
+
blanket
|
133 |
sculpture
|
134 |
+
hood
|
135 |
sconce
|
136 |
vase
|
137 |
traffic light
|
138 |
tray
|
139 |
+
ashcan
|
140 |
fan
|
141 |
pier
|
142 |
+
crt screen
|
143 |
plate
|
144 |
monitor
|
145 |
bulletin board
|
146 |
shower
|
147 |
radiator
|
148 |
+
glass
|
149 |
clock
|
150 |
flag
|