kiselyovd
/

brain-mri-segmentation

Image Segmentation

medical-imaging

semantic-segmentation

Model card Files Files and versions

kiselyovd commited on Jun 25

Commit

47775eb

·

verified ·

1 Parent(s): c656b45

fix: add id2label/label2id for binary segmentation (background/tumor)

Files changed (1) hide show

config.json +10 -1

config.json CHANGED Viewed

@@ -65,5 +65,14 @@
     2,
     2
   ],
-  "transformers_version": "5.5.4"
 }

     2,
     2
   ],
+  "transformers_version": "5.5.4",
+  "num_labels": 2,
+  "id2label": {
+    "0": "background",
+    "1": "tumor"
+  },
+  "label2id": {
+    "background": 0,
+    "tumor": 1
+  }
 }