milschla
/

mushroomclassifier

Image Classification

Model card Files Files and versions

milschla commited on Jun 20, 2023

Commit

a0cf0ed

·

1 Parent(s): 3784bf3

Update config.json

Files changed (1) hide show

config.json +20 -22

config.json CHANGED Viewed

@@ -1,31 +1,29 @@
 {
   "architectures": [
-    "mobilenetForImageClassification"
   ],
   "downsample_in_first_stage": false,
   "id2label": {
-    "0": "asdf",
-    "1": "dfgh",
-    "2": "fghj",
-    "3": "gjhk",
-    "4": "yxcv",
-    "5": "cvbn",
-    "6": "bnme",
-    "7": "sad",
-    "8": "hen"
   },
   "label2id": {
-    "asdf": "0",
-    "dfgh": "1",
-    "fghj": "2",
-    "gjhk": "3",
-    "yxcv": "4",
-    "cvbn": "5",
-    "bnme": "6",
-    "sad": "7",
-    "hen": "8",
   },
-  "model_type": "mobilenet",
-  "num_channels": 3,
-  "transformers_version": "4.18.0.dev0"
 }

 {
   "architectures": [
+    "MushroomClassifier"
   ],
   "downsample_in_first_stage": false,
   "id2label": {
+    "0": "Agaricus",
+    "1": "Amanita",
+    "2": "Boletus",
+    "3": "Cortinarius",
+    "4": "Entoloma",
+    "5": "Hygrocybe",
+    "6": "Lactarius",
+    "7": "Russula",
+    "8": "Suillus"
   },
   "label2id": {
+    "Agaricus": "0",
+    "Amanita": "1",
+    "Boletus": "2",
+    "Cortinarius": "3",
+    "Entoloma": "4",
+    "Hygrocybe": "5",
+    "Lactarius": "6",
+    "Russula": "7",
+    "Suillus": "8",
   },
+  "num_channels": 9,
 }