Upload 8 files

README.md

@@ -1,8 +1,8 @@
 ---
-title: Segmentation
-emoji: 👀
-colorFrom: red
-colorTo: blue
+title: Non32
+emoji: 🚀
+colorFrom: gray
+colorTo: purple
 sdk: gradio
 sdk_version: 3.44.4
 app_file: app.py

app.py

@@ -8,23 +8,23 @@ import tensorflow as tf
 from transformers import SegformerFeatureExtractor, TFSegformerForSemanticSegmentation
 
 feature_extractor = SegformerFeatureExtractor.from_pretrained(
-    "nvidia/segformer-b5-finetuned-ade-640-640"
+    "nvidia/segformer-b5-finetuned-cityscapes-1024-1024"
 )
 model = TFSegformerForSemanticSegmentation.from_pretrained(
-    "nvidia/segformer-b5-finetuned-ade-640-640"
+    "nvidia/segformer-b5-finetuned-cityscapes-1024-1024"
 )
 
 def ade_palette():
     """ADE20K palette that maps each class to RGB values."""
     return [
-        [204, 87, 92],
-        [112, 185, 212],
-        [45, 189, 106],
-        [234, 123, 67],
-        [78, 56, 123],
-        [210, 32, 89],
-        [90, 180, 56],
-        [155, 102, 200],
+        [0, 0, 0],
+        [255, 0, 0],
+        [255, 255, 0],
+        [255, 255, 255],
+        [255, 0, 255],
+        [0, 255, 0],
+        [0, 255, 255],
+        [0, 0, 255],
         [33, 147, 176],
         [255, 183, 76],
         [67, 123, 89],
@@ -235,7 +235,7 @@ def sepia(input_img):
 demo = gr.Interface(fn=sepia,
                     inputs=gr.Image(shape=(400, 600)),
                     outputs=['plot'],
-                    examples=["ADE_val_00000001.jpeg", "ADE_val_00001159.jpg", "ADE_val_00001248.jpg", "ADE_val_00001472.jpg"],
+                    examples=["cheonggyecheon_stream_in_seoul_city.jpg", "Incheon_stadium.jpg", "Incheon_city.jpg"],
                     allow_flagging='never')
 
 

labels.txt

@@ -1,150 +1,19 @@
-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
+building
+wall
 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
+traffic sign
+vegetation
+terrain
+sky
+person
+rider
+car
+truck
+bus
+train
+motorcycle
+bicycle