Upload SegformerForSemanticSegmentation

README.md

@@ -1,6 +1,8 @@
 ---
 library_name: transformers
-tags: []
+tags:
+- vision
+- image-segmentation
 ---
 
 # Model Card for Model ID

config.json

@@ -29,37 +29,29 @@
   ],
   "id2label": {
     "0": "Ignore",
-    "1": "Major_Road",
-    "2": "Flyover",
-    "3": "Small_Road",
-    "4": "Street",
-    "5": "Skyscraper",
-    "6": "Large_Building",
-    "7": "Mid_Building",
-    "8": "Small_Building",
-    "9": "Commercial",
-    "10": "Parking",
-    "11": "Waterway",
-    "12": "Swimming_Pool",
-    "13": "Greenery"
+    "1": "Background",
+    "2": "Building",
+    "3": "Bridge/Flyover",
+    "4": "Highway/Major_Road",
+    "5": "Street/Lane",
+    "6": "Water",
+    "7": "Barren",
+    "8": "Forest",
+    "9": "Agricultural"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "label2id": {
-    "Commercial": 9,
-    "Flyover": 2,
-    "Greenery": 13,
+    "Agricultural": 9,
+    "Background": 1,
+    "Barren": 7,
+    "Bridge/Flyover": 3,
+    "Building": 2,
+    "Forest": 8,
+    "Highway/Major_Road": 4,
     "Ignore": 0,
-    "Large_Building": 6,
-    "Major_Road": 1,
-    "Mid_Building": 7,
-    "Parking": 10,
-    "Skyscraper": 5,
-    "Small_Building": 8,
-    "Small_Road": 3,
-    "Street": 4,
-    "Swimming_Pool": 12,
-    "Waterway": 11
+    "Street/Lane": 5,
+    "Water": 6
   },
   "layer_norm_eps": 1e-06,
   "mlp_ratios": [

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5a7a0b266c6d67bca8274c63af90a89ca880e5979c47dc141eb523dc9336f221
-size 109480936
+oid sha256:4ead252806c3594ba63931c2cf7ac82b2451efc44135deb3899ae6c6578a122b
+size 109468632