Hugging Face's logo

Spaces:
0jung
/
Paper3_Segmentation
Runtime error

App Files Community
0jung commited on
Commit
32bb6b5
•
1 Parent(s): ee173f9

modify

Browse files
Files changed (2) hide show
  1. app.py +2 -1
  2. labels.txt +53 -52
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import MaskFormerImageProcessor, MaskFormerForInstanceSegmenta
5
 
6
  # Load the model using TensorFlow
7
  model = MaskFormerForInstanceSegmentation.from_pretrained(
8
- "nvidia/segformer-b0-finetuned-cityscapes-1024-1024"
9
  )
10
 
11
  def ade_palette():
@@ -160,6 +160,7 @@ def ade_palette():
160
  [99, 28, 210],
161
  [189, 78, 47],
162
  [28, 210, 99],
 
163
  ]
164
 
165
  labels_list = []
 
5
 
6
  # Load the model using TensorFlow
7
  model = MaskFormerForInstanceSegmentation.from_pretrained(
8
+ "facebook/maskformer-swin-large-ade"
9
  )
10
 
11
  def ade_palette():
 
160
  [99, 28, 210],
161
  [189, 78, 47],
162
  [28, 210, 99],
163
+ [26, 165, 222],
164
  ]
165
 
166
  labels_list = []
labels.txt CHANGED
@@ -4,23 +4,23 @@ sky
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,94 +32,95 @@ armchair
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,22 +129,22 @@ bicycle
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
142
  plate
143
  monitor
144
  bulletin board
145
  shower
146
  radiator
147
- glass
148
  clock
149
  flag
 
4
  floor
5
  tree
6
  ceiling
7
+ road, route
8
  bed
9
+ window
10
  grass
11
  cabinet
12
+ sidewalk, pavement
13
  person
14
+ earth, ground
15
  door
16
  table
17
+ mountain, mount
18
  plant
19
  curtain
20
  chair
21
  car
22
  water
23
+ painting, picture
24
  sofa
25
  shelf
26
  house
 
32
  seat
33
  fence
34
  desk
35
+ rock, stone
36
+ wardrobe, closet, press
37
  lamp
38
+ tub
39
+ rail
40
  cushion
41
+ base, pedestal, stand
42
+ box
43
+ column, pillar
44
+ signboard, sign
45
+ chest of drawers, chest, bureau, dresser
46
  counter
47
  sand
48
  sink
49
  skyscraper
50
  fireplace
51
+ refrigerator, icebox
52
+ grandstand, covered stand
53
  path
54
  stairs
55
  runway
56
+ case, display case, showcase, vitrine
57
+ pool table, billiard table, snooker table
58
  pillow
59
+ screen door, screen
60
+ stairway, staircase
61
  river
62
+ bridge, span
63
  bookcase
64
+ blind, screen
65
  coffee table
66
+ toilet, can, commode, crapper, pot, potty, stool, throne
67
  flower
68
  book
69
  hill
70
  bench
71
  countertop
72
  stove
73
+ palm, palm tree
74
  kitchen island
75
  computer
76
  swivel chair
77
  boat
78
  bar
79
  arcade machine
80
+ hovel, hut, hutch, shack, shanty
81
  bus
82
  towel
83
  light
84
  truck
85
  tower
86
  chandelier
87
+ awning, sunshade, sunblind
88
+ street lamp
89
  booth
90
+ TV
91
+ plane
92
  dirt track
93
+ clothes
94
  pole
95
+ land, ground, soil
96
+ bannister, banister, balustrade, balusters, handrail
97
+ escalator, moving staircase, moving stairway
98
+ ottoman, pouf, pouffe, puff, hassock
99
  bottle
100
+ buffet, counter, sideboard
101
+ poster, posting, placard, notice, bill, card
102
  stage
103
  van
104
  ship
105
  fountain
106
+ conveyer belt, conveyor belt, conveyer, conveyor, transporter
107
  canopy
108
+ washer, automatic washer, washing machine
109
+ plaything, toy
110
+ pool
111
  stool
112
+ barrel, cask
113
+ basket, handbasket
114
+ falls
115
  tent
116
  bag
117
+ minibike, motorbike
118
  cradle
119
  oven
120
  ball
121
+ food, solid food
122
+ step, stair
123
+ tank, storage tank
124
  trade name
125
  microwave
126
  pot
 
129
  lake
130
  dishwasher
131
  screen
132
+ blanket, cover
133
  sculpture
134
+ hood, exhaust hood
135
  sconce
136
  vase
137
  traffic light
138
  tray
139
+ trash can
140
  fan
141
  pier
142
+ CRT screen
143
  plate
144
  monitor
145
  bulletin board
146
  shower
147
  radiator
148
+ glass, drinking glass
149
  clock
150
  flag