modify

app.py

@@ -10,10 +10,10 @@ import requests
 
 
 feature_extractor = MaskFormerImageProcessor.from_pretrained(
-    "facebook/maskformer-swin-large-ade"
+    "nvidia/segformer-b5-finetuned-cityscapes-1024-1024"
 )
 model = MaskFormerForInstanceSegmentation.from_pretrained(
-    "facebook/maskformer-swin-large-ade"
+    "nvidia/segformer-b5-finetuned-cityscapes-1024-1024"
 )
 
 def ade_palette():
@@ -168,7 +168,7 @@ def ade_palette():
         [99, 28, 210],
         [189, 78, 47],
         [28, 210, 99],
-        [141, 21, 43]
+        [141, 21, 43],
     ]
 
 labels_list = []

labels.txt

@@ -4,23 +4,23 @@ sky
 floor
 tree
 ceiling
-road, route
+road
 bed
-window
+windowpane
 grass
 cabinet
-sidewalk, pavement
+sidewalk
 person
-earth, ground
+earth
 door
 table
-mountain, mount
+mountain
 plant
 curtain
 chair
 car
 water
-painting, picture
+painting
 sofa
 shelf
 house
@@ -32,95 +32,95 @@ armchair
 seat
 fence
 desk
-rock, stone
-wardrobe, closet, press
+rock
+wardrobe
 lamp
-tub
-rail
+bathtub
+railing
 cushion
-base, pedestal, stand
+base
 box
-column, pillar
-signboard, sign
-chest of drawers, chest, bureau, dresser
+column
+signboard
+chest of drawers
 counter
 sand
 sink
 skyscraper
 fireplace
-refrigerator, icebox
-grandstand, covered stand
+refrigerator
+grandstand
 path
 stairs
 runway
-case, display case, showcase, vitrine
-pool table, billiard table, snooker table
+case
+pool table
 pillow
-screen door, screen
-stairway, staircase
+screen door
+stairway
 river
-bridge, span
+bridge
 bookcase
-blind, screen
+blind
 coffee table
-toilet, can, commode, crapper, pot, potty, stool, throne
+toilet
 flower
 book
 hill
 bench
 countertop
 stove
-palm, palm tree
+palm
 kitchen island
 computer
 swivel chair
 boat
 bar
 arcade machine
-hovel, hut, hutch, shack, shanty
+hovel
 bus
 towel
 light
 truck
 tower
 chandelier
-awning, sunshade, sunblind
-street lamp
+awning
+streetlight
 booth
-TV
-plane
+television receiver
+airplane
 dirt track
-clothes
+apparel
 pole
-land, ground, soil
-bannister, banister, balustrade, balusters, handrail
-escalator, moving staircase, moving stairway
-ottoman, pouf, pouffe, puff, hassock
+land
+bannister
+escalator
+ottoman
 bottle
-buffet, counter, sideboard
-poster, posting, placard, notice, bill, card
+buffet
+poster
 stage
 van
 ship
 fountain
-conveyer belt, conveyor belt, conveyer, conveyor, transporter
+conveyer belt
 canopy
-washer, automatic washer, washing machine
-plaything, toy
-pool
+washer
+plaything
+swimming pool
 stool
-barrel, cask
-basket, handbasket
-falls
+barrel
+basket
+waterfall
 tent
 bag
-minibike, motorbike
+minibike
 cradle
 oven
 ball
-food, solid food
-step, stair
-tank, storage tank
+food
+step
+tank
 trade name
 microwave
 pot
@@ -129,22 +129,22 @@ bicycle
 lake
 dishwasher
 screen
-blanket, cover
+blanket
 sculpture
-hood, exhaust hood
+hood
 sconce
 vase
 traffic light
 tray
-trash can
+ashcan
 fan
 pier
-CRT screen
+crt screen
 plate
 monitor
 bulletin board
 shower
 radiator
-glass, drinking glass
+glass
 clock
 flag