Hugging Face's logo

Spaces:
0jung
/
Paper3_Segmentation
Runtime error

App Files Community
김영중 commited on
Commit
9bfde87
·
1 Parent(s): f9d2928

modify segmentation

Browse files
Files changed (2) hide show
  1. app.py +4 -5
  2. 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, AutoModelForImageSegmentation, DetrConfig
9
 
10
 
11
- feature_extractor = AutoModelForImageSegmentation.from_pretrained(
12
  "facebook/maskformer-swin-large-ade"
13
  )
14
- config = DetrConfig.from_pretrained("facebook/maskformer-swin-large-ade")
15
- model = AutoModelForImageSegmentation.from_pretrained(
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, 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,94 +32,94 @@ armchair
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
  sink
48
  skyscraper
49
  fireplace
50
- refrigerator, icebox
51
- grandstand, covered stand
52
  path
53
  stairs
54
  runway
55
- case, display case, showcase, vitrine
56
- pool table, billiard table, snooker table
57
  pillow
58
- screen door, screen
59
- stairway, staircase
60
  river
61
- bridge, span
62
  bookcase
63
- blind, screen
64
  coffee table
65
- toilet, can, commode, crapper, pot, potty, stool, throne
66
  flower
67
  book
68
  hill
69
  bench
70
  countertop
71
  stove
72
- palm, palm tree
73
  kitchen island
74
  computer
75
  swivel chair
76
  boat
77
  bar
78
  arcade machine
79
- hovel, hut, hutch, shack, shanty
80
  bus
81
  towel
82
  light
83
  truck
84
  tower
85
  chandelier
86
- awning, sunshade, sunblind
87
- street lamp
88
  booth
89
- tv
90
- plane
91
  dirt track
92
- clothes
93
  pole
94
- land, ground, soil
95
- bannister, banister, balustrade, balusters, handrail
96
- escalator, moving staircase, moving stairway
97
- ottoman, pouf, pouffe, puff, hassock
98
  bottle
99
- buffet, counter, sideboard
100
- poster, posting, placard, notice, bill, card
101
  stage
102
  van
103
  ship
104
  fountain
105
- conveyer belt, conveyor belt, conveyer, conveyor, transporter
106
  canopy
107
- washer, automatic washer, washing machine
108
- plaything, toy
109
- pool
110
  stool
111
- barrel, cask
112
- basket, handbasket
113
- falls
114
  tent
115
  bag
116
- minibike, motorbike
117
  cradle
118
  oven
119
  ball
120
- food, solid food
121
- step, stair
122
- tank, storage tank
123
  trade name
124
  microwave
125
  pot
@@ -128,14 +128,14 @@ bicycle
128
  lake
129
  dishwasher
130
  screen
131
- blanket, cover
132
  sculpture
133
- hood, exhaust hood
134
  sconce
135
  vase
136
  traffic light
137
  tray
138
- trash can
139
  fan
140
  pier
141
  crt screen
@@ -144,6 +144,6 @@ monitor
144
  bulletin board
145
  shower
146
  radiator
147
- glass, drinking 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