mushroomclassifier / config.json
milschla's picture
Update config.json
dfc0144
{
"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
}