Spaces:

gch0301
/

sixthspace_free

Sleeping

App Files Files Community

gch0301 commited on Nov 9

Commit

dffaf20

1 Parent(s): 8307dc4

1

Browse files

Files changed (2) hide show

app.py +1 -10
labels.txt +150 -197

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from PIL import Image
 import torch
 from transformers import AutoImageProcessor, AutoModelForSemanticSegmentation
-MODEL_ID = "ZhengPeng7/BiRefNet"
 processor = AutoImageProcessor.from_pretrained(MODEL_ID)
 model = AutoModelForSemanticSegmentation.from_pretrained(MODEL_ID)
@@ -37,15 +37,6 @@ def ade_palette():
         [28, 210, 99], [78, 89, 189], [189, 78, 57], [112, 200, 78], [189, 47, 78], [205, 112, 57],
         [78, 145, 57], [200, 78, 112], [99, 89, 145], [200, 156, 78], [57, 78, 145], [78, 57, 99],
         [57, 78, 145], [145, 112, 78], [78, 89, 145], [210, 99, 28], [145, 78, 189], [57, 200, 136],
-        [89, 156, 78], [145, 78, 99], [99, 28, 210], [189, 78, 47], [28, 210, 99], [78, 145, 57],[154, 87, 92], [112, 185, 212], [45, 189, 106], [234, 123, 67], [78, 56, 123], [210, 32, 89],
-        [40, 180, 56], [105, 102, 200], [0, 147, 176], [205, 183, 76], [17, 123, 89], [140, 60, 45],
-        [84, 112, 200], [6, 45, 189], [150, 56, 123], [37, 92, 204], [70, 56, 123], [0, 78, 123],
-        [106, 200, 56], [0, 90, 210], [6, 123, 67], [130, 56, 123], [73, 67, 45], [0, 134, 200],
-        [17, 56, 123], [28, 123, 67], [0, 210, 90], [0, 56, 189], [73, 56, 123], [56, 106, 200],
-        [139, 56, 45], [112, 200, 6], [56, 73, 45], [150, 32, 90], [123, 45, 28], [150, 156, 56],
-        [45, 17, 123], [56, 45, 28], [45, 6, 123], [73, 67, 56], [56, 78, 73], [160, 90, 32],
-        [123, 56, 139], [45, 150, 134], [67, 73, 56], [73, 45, 67], [90, 32, 160], [150, 45, 78],
-        [32, 210, 40],
     ]
 labels_list = []

 import torch
 from transformers import AutoImageProcessor, AutoModelForSemanticSegmentation
+MODEL_ID = "Xenova/segformer-b0-finetuned-ade-512-512"
 processor = AutoImageProcessor.from_pretrained(MODEL_ID)
 model = AutoModelForSemanticSegmentation.from_pretrained(MODEL_ID)
         [28, 210, 99], [78, 89, 189], [189, 78, 57], [112, 200, 78], [189, 47, 78], [205, 112, 57],
         [78, 145, 57], [200, 78, 112], [99, 89, 145], [200, 156, 78], [57, 78, 145], [78, 57, 99],
         [57, 78, 145], [145, 112, 78], [78, 89, 145], [210, 99, 28], [145, 78, 189], [57, 200, 136],
     ]
 labels_list = []

labels.txt CHANGED Viewed

@@ -1,197 +1,150 @@
-Airplane,
-Ant,
-Antenna,
-Archery,
-Axe,
-BabyCarriage,
-Bag,
-BalanceBeam,
-Balcony,
-Balloon,
-Basket,
-BasketballHoop,
-Beatle,
-Bed,
-Bee,
-Bench,
-Bicycle
-BicycleFrame,
-BicycleStand,
-Boat,
-Bonsai,
-BoomLift,
-Bridge,
-BunkBed,
-Butterfly, Button, Cable, CableLift, Cage, Camcorder, Cannon, Canoe, Car,
-CarParkDropArm,
-Carriage, Cart, Caterpillar, CeilingLamp, Centipede, Chair, Clip, Clock, Clothes, CoatHanger, Comb, ConcretePumpTruck, Crack, Crane, '
-Cup,
-DentalChair,
-Desk,
-DeskChair,
-Diagram,
-DishRack,
-DoorHandle,
-Dragonfish,
-Dragonfly,
-Drum,
-Earphone,
-Easel,
-ElectricIron,
-Excavator,
-Eyeglasses, '
-Fan,
-Fence,
-Fencing,
-FerrisWheel,
-FireExtinguisher,
-Fishing,
-Flag,
-FloorLamp,
-Forklift,
-GasStation,
-Gate,
-Gear,
-Goal,
-Golf,
-GymEquipment,
-Hammock, '
-Handcart,
-Handcraft,
-Handrail,
-HangGlider,
-Harp, Harvester,
-Headset,
-Helicopter,
-Helmet,
-Hook,
-HorizontalBar,
-Hydrovalve,
-IroningTable,
-Jewelry,
-Key, '
-KidsPlayground,
-Kitchenware,
-Kite,
-Knife,
-Ladder,
-LaundryRack,
-Lightning,
-Lobster,
-Locust,
-Machine,
-MachineGun,
-MagazineRack,
-Mantis,
-Medal,
-MemorialArchway, '
-Microphone,
-Missile,
-MobileHolder,
-Monitor,
-Mosquito,
-Motorcycle,
-MovingTrolley,
-Mower,
-MusicPlayer,
-MusicStand,
-ObservationTower,
-Octopus,
-OilWell, '
-OlympicLogo,
-OperatingTable,
-OutdoorFitnessEquipment,
-Parachute,
-Pavilion,
-Piano,
-Pipe,
-PlowHarrow,
-PoleVault,
-Punchbag,
-Rack,
-Racket,
-Rifle,
-Ring,
-Robot, '
-RockClimbing,
-Rope,
-Sailboat,
-Satellite,
-Scaffold,
-Scale,
-Scissor,
-Scooter,
-Sculpture,
-Seadragon,
-Seahorse,
-Seal,
-SewingMachine,
-Ship,
-Shoe,
-ShoppingCart, '
-ShoppingTrolley,
-Shower,
-Shrimp,
-Signboard,
-Skateboarding,
-Skeleton,
-Skiing,
-Spade,
-SpeedBoat,
-Spider,
-Spoon,
-Stair,
-Stand,
-Stationary,
-SteeringWheel, '
-Stethoscope,
-Stool,
-Stove,
-StreetLamp,
-SweetStand,
-Swing,
-Sword,
-TV,
-Table,
-TableChair,
-TableLamp,
-TableTennis,
-Tank,
-Tapeline,
-Teapot,
-Telescope,
-Tent, '
-TobaccoPipe,
-Toy,
-Tractor,
-TrafficLight,
-TrafficSign,
-Trampoline,
-TransmissionTower,
-Tree,
-Tricycle,
-TrimmerCover,
-Tripod,
-Trombone,
-Truck,
-Trumpet,
-Tuba, '
-UAV,
-Umbrella,
-UnevenBars,
-UtilityPole,
-VacuumCleaner,
-Violin,
-Wakesurfing,
-Watch,
-WaterTower,
-WateringPot,
-Well,
-WellLid,
-Wheel,
-Wheelchair,
-WindTurbine,
-Windmill,
-WineGlass,
-WireWhisk,
-Yacht'

+wall
+building
+sky
+floor
+tree
+ceiling
+road
+bed
+windowpane
+grass
+cabinet
+sidewalk
+person
+earth
+door
+table
+mountain
+plant
+curtain
+chair
+car
+water
+painting
+sofa
+shelf
+house
+sea
+mirror
+rug
+field
+armchair
+seat
+fence
+desk
+rock
+wardrobe
+lamp
+bathtub
+railing
+cushion
+base
+box
+column
+signboard
+chest of drawers
+counter
+sand
+sink
+skyscraper
+fireplace
+refrigerator
+grandstand
+path
+stairs
+runway
+case
+pool table
+pillow
+screen door
+stairway
+river
+bridge
+bookcase
+blind
+coffee table
+toilet
+flower
+book
+hill
+bench
+countertop
+stove
+palm
+kitchen island
+computer
+swivel chair
+boat
+bar
+arcade machine
+hovel
+bus
+towel
+light
+truck
+tower
+chandelier
+awning
+streetlight
+booth
+television receiver
+airplane
+dirt track
+apparel
+pole
+land
+bannister
+escalator
+ottoman
+bottle
+buffet
+poster
+stage
+van
+ship
+fountain
+conveyer belt
+canopy
+washer
+plaything
+swimming pool
+stool
+barrel
+basket
+waterfall
+tent
+bag
+minibike
+cradle
+oven
+ball
+food
+step
+tank
+trade name
+microwave
+pot
+animal
+bicycle
+lake
+dishwasher
+screen
+blanket
+sculpture
+hood
+sconce
+vase
+traffic light
+tray
+ashcan
+fan
+pier
+crt screen
+plate
+monitor
+bulletin board
+shower
+radiator
+glass
+clock
+flag