nvidia
/

segformer-b0-finetuned-ade-512-512

Image Segmentation

Inference Endpoints

Model card Files Files and versions Community

nielsr HF staff commited on Oct 15, 2021

Commit

77fa31a

•

1 Parent(s): 409aa62

Update model

Files changed (2) hide show

config.json +3 -2
preprocessor_config.json +1 -1

config.json CHANGED Viewed

@@ -1,6 +1,6 @@
 {
   "architectures": [
-    "SegFormerForImageSegmentation"
   ],
   "attention_probs_dropout_prob": 0.0,
   "classifier_dropout_prob": 0.1,
@@ -354,6 +354,7 @@
     3,
     3
   ],
   "sr_ratios": [
     8,
     4,
@@ -367,5 +368,5 @@
     2
   ],
   "torch_dtype": "float32",
-  "transformers_version": "4.10.0.dev0"
 }

 {
   "architectures": [
+    "SegformerForImageSegmentation"
   ],
   "attention_probs_dropout_prob": 0.0,
   "classifier_dropout_prob": 0.1,
     3,
     3
   ],
+  "reshape_last_stage": true,
   "sr_ratios": [
     8,
     4,
     2
   ],
   "torch_dtype": "float32",
+  "transformers_version": "4.12.0.dev0"
 }

preprocessor_config.json CHANGED Viewed

@@ -8,7 +8,7 @@
   "do_pad": true,
   "do_random_crop": false,
   "do_resize": true,
-  "feature_extractor_type": "SegFormerFeatureExtractor",
   "image_mean": [
     0.485,
     0.456,

   "do_pad": true,
   "do_random_crop": false,
   "do_resize": true,
+  "feature_extractor_type": "SegformerFeatureExtractor",
   "image_mean": [
     0.485,
     0.456,