nahuelnb's picture
Upload config.json with huggingface_hub
3965625 verified
{
"architectures": [
"PlantPathologyModel"
],
"model_type": "efficientnet",
"num_classes": 4,
"image_size": 260,
"id2label": {
"0": "healthy",
"1": "multiple_diseases",
"2": "rust",
"3": "scab"
},
"label2id": {
"healthy": 0,
"multiple_diseases": 1,
"rust": 2,
"scab": 3
},
"backbone": "efficientnet_b2",
"pretrained_backbone": true,
"dropout": 0.3,
"normalization": {
"mean": [
0.485,
0.456,
0.406
],
"std": [
0.229,
0.224,
0.225
]
}
}