Spaces:
Runtime error
Runtime error
김영중
commited on
Commit
•
9bfde87
1
Parent(s):
f9d2928
modify segmentation
Browse files- app.py +4 -5
- labels.txt +52 -52
app.py
CHANGED
@@ -5,15 +5,14 @@ import matplotlib.pyplot as plt
|
|
5 |
import numpy as np
|
6 |
from PIL import Image
|
7 |
import tensorflow as tf
|
8 |
-
from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
|
9 |
|
10 |
|
11 |
-
feature_extractor =
|
12 |
"facebook/maskformer-swin-large-ade"
|
13 |
)
|
14 |
-
|
15 |
-
|
16 |
-
"facebook/maskformer-swin-large-ade", config=config
|
17 |
)
|
18 |
|
19 |
def ade_palette():
|
|
|
5 |
import numpy as np
|
6 |
from PIL import Image
|
7 |
import tensorflow as tf
|
8 |
+
from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
|
9 |
|
10 |
|
11 |
+
feature_extractor = SegformerFeatureExtractor.from_pretrained(
|
12 |
"facebook/maskformer-swin-large-ade"
|
13 |
)
|
14 |
+
model = TFSegformerForSemanticSegmentation.from_pretrained(
|
15 |
+
"facebook/maskformer-swin-large-ade"
|
|
|
16 |
)
|
17 |
|
18 |
def ade_palette():
|
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,94 +32,94 @@ armchair
|
|
32 |
seat
|
33 |
fence
|
34 |
desk
|
35 |
-
rock
|
36 |
-
wardrobe
|
37 |
lamp
|
38 |
-
|
39 |
-
|
40 |
cushion
|
41 |
-
base
|
42 |
-
|
43 |
-
|
44 |
-
|
45 |
-
chest of drawers, chest, bureau, dresser
|
46 |
counter
|
|
|
47 |
sink
|
48 |
skyscraper
|
49 |
fireplace
|
50 |
-
refrigerator
|
51 |
-
grandstand
|
52 |
path
|
53 |
stairs
|
54 |
runway
|
55 |
-
case
|
56 |
-
pool table
|
57 |
pillow
|
58 |
-
screen door
|
59 |
-
stairway
|
60 |
river
|
61 |
-
bridge
|
62 |
bookcase
|
63 |
-
blind
|
64 |
coffee table
|
65 |
-
toilet
|
66 |
flower
|
67 |
book
|
68 |
hill
|
69 |
bench
|
70 |
countertop
|
71 |
stove
|
72 |
-
palm
|
73 |
kitchen island
|
74 |
computer
|
75 |
swivel chair
|
76 |
boat
|
77 |
bar
|
78 |
arcade machine
|
79 |
-
hovel
|
80 |
bus
|
81 |
towel
|
82 |
light
|
83 |
truck
|
84 |
tower
|
85 |
chandelier
|
86 |
-
awning
|
87 |
-
|
88 |
booth
|
89 |
-
|
90 |
-
|
91 |
dirt track
|
92 |
-
|
93 |
pole
|
94 |
-
land
|
95 |
-
bannister
|
96 |
-
escalator
|
97 |
-
ottoman
|
98 |
bottle
|
99 |
-
buffet
|
100 |
-
poster
|
101 |
stage
|
102 |
van
|
103 |
ship
|
104 |
fountain
|
105 |
-
conveyer belt
|
106 |
canopy
|
107 |
-
washer
|
108 |
-
plaything
|
109 |
-
pool
|
110 |
stool
|
111 |
-
barrel
|
112 |
-
basket
|
113 |
-
|
114 |
tent
|
115 |
bag
|
116 |
-
minibike
|
117 |
cradle
|
118 |
oven
|
119 |
ball
|
120 |
-
food
|
121 |
-
step
|
122 |
-
tank
|
123 |
trade name
|
124 |
microwave
|
125 |
pot
|
@@ -128,14 +128,14 @@ bicycle
|
|
128 |
lake
|
129 |
dishwasher
|
130 |
screen
|
131 |
-
blanket
|
132 |
sculpture
|
133 |
-
hood
|
134 |
sconce
|
135 |
vase
|
136 |
traffic light
|
137 |
tray
|
138 |
-
|
139 |
fan
|
140 |
pier
|
141 |
crt screen
|
@@ -144,6 +144,6 @@ monitor
|
|
144 |
bulletin board
|
145 |
shower
|
146 |
radiator
|
147 |
-
glass
|
148 |
clock
|
149 |
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 |
+
column
|
43 |
+
signboard
|
44 |
+
chest of drawers
|
|
|
45 |
counter
|
46 |
+
sand
|
47 |
sink
|
48 |
skyscraper
|
49 |
fireplace
|
50 |
+
refrigerator
|
51 |
+
grandstand
|
52 |
path
|
53 |
stairs
|
54 |
runway
|
55 |
+
case
|
56 |
+
pool table
|
57 |
pillow
|
58 |
+
screen door
|
59 |
+
stairway
|
60 |
river
|
61 |
+
bridge
|
62 |
bookcase
|
63 |
+
blind
|
64 |
coffee table
|
65 |
+
toilet
|
66 |
flower
|
67 |
book
|
68 |
hill
|
69 |
bench
|
70 |
countertop
|
71 |
stove
|
72 |
+
palm
|
73 |
kitchen island
|
74 |
computer
|
75 |
swivel chair
|
76 |
boat
|
77 |
bar
|
78 |
arcade machine
|
79 |
+
hovel
|
80 |
bus
|
81 |
towel
|
82 |
light
|
83 |
truck
|
84 |
tower
|
85 |
chandelier
|
86 |
+
awning
|
87 |
+
streetlight
|
88 |
booth
|
89 |
+
television receiver
|
90 |
+
airplane
|
91 |
dirt track
|
92 |
+
apparel
|
93 |
pole
|
94 |
+
land
|
95 |
+
bannister
|
96 |
+
escalator
|
97 |
+
ottoman
|
98 |
bottle
|
99 |
+
buffet
|
100 |
+
poster
|
101 |
stage
|
102 |
van
|
103 |
ship
|
104 |
fountain
|
105 |
+
conveyer belt
|
106 |
canopy
|
107 |
+
washer
|
108 |
+
plaything
|
109 |
+
swimming pool
|
110 |
stool
|
111 |
+
barrel
|
112 |
+
basket
|
113 |
+
waterfall
|
114 |
tent
|
115 |
bag
|
116 |
+
minibike
|
117 |
cradle
|
118 |
oven
|
119 |
ball
|
120 |
+
food
|
121 |
+
step
|
122 |
+
tank
|
123 |
trade name
|
124 |
microwave
|
125 |
pot
|
|
|
128 |
lake
|
129 |
dishwasher
|
130 |
screen
|
131 |
+
blanket
|
132 |
sculpture
|
133 |
+
hood
|
134 |
sconce
|
135 |
vase
|
136 |
traffic light
|
137 |
tray
|
138 |
+
ashcan
|
139 |
fan
|
140 |
pier
|
141 |
crt screen
|
|
|
144 |
bulletin board
|
145 |
shower
|
146 |
radiator
|
147 |
+
glass
|
148 |
clock
|
149 |
flag
|