Upload SegformerForSemanticSegmentation

README.md

@@ -1,6 +1,6 @@
 ---
-library_name: transformers
 license: apache-2.0
+library_name: transformers
 datasets:
 - pruhtopia/mangrove-loss-dataset
 pipeline_tag: image-segmentation

config.json

@@ -1,16 +1,16 @@
 {
-  "_name_or_path": "nvidia/mit-b0",
+  "_name_or_path": "nvidia/mit-b5",
   "architectures": [
     "SegformerForSemanticSegmentation"
   ],
   "attention_probs_dropout_prob": 0.0,
   "classifier_dropout_prob": 0.1,
-  "decoder_hidden_size": 256,
+  "decoder_hidden_size": 768,
   "depths": [
-    2,
-    2,
-    2,
-    2
+    3,
+    6,
+    40,
+    3
   ],
   "downsampling_rates": [
     1,
@@ -22,20 +22,20 @@
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
   "hidden_sizes": [
-    32,
     64,
-    160,
-    256
+    128,
+    320,
+    512
   ],
   "id2label": {
-    "0": "background",
-    "1": "loss"
+    "0": "nomangrove",
+    "1": "mangrove"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "label2id": {
-    "background": 0,
-    "loss": 1
+    "mangrove": 1,
+    "nomangrove": 0
   },
   "layer_norm_eps": 1e-06,
   "mlp_ratios": [

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:db43ce7813f6e8c77ff6a01f8dd81429d52550a9b8d7b1b3b8e7ed9c2e204557
-size 14884776
+oid sha256:18e8212361a3eb79711a98f5d60caff582722e47c761ea6db50242affa2d5772
+size 338603704