plant-classifier-39 / model_config.json
f4m1's picture
Add classifier metadata and model card
5125faf verified
Raw
History Blame Contribute Delete
472 Bytes
{
"architecture": "torchvision.models.efficientnet_b3",
"checkpoint_file": "model.pt",
"state_dict_key": "model",
"class_mapping_key": "class_to_idx",
"num_classes": 39,
"image_size": 380,
"resize": "longest_side",
"preserve_aspect_ratio": true,
"padding": "center_black_to_square",
"color_mode": "RGB",
"normalization_mean": [0.485, 0.456, 0.406],
"normalization_std": [0.229, 0.224, 0.225],
"confidence": "raw_softmax",
"calibrated": false
}