dental-impression-classifier / model_config.json
mzou1223's picture
Add Model configuration
6a6c80b verified
raw
history blame contribute delete
304 Bytes
{
"model_type": "efficientnet_b0",
"num_classes": 3,
"class_names": [
"bad",
"borderline",
"good"
],
"img_size": 224,
"test_accuracy": 60.97560975609756,
"best_val_accuracy": 65.0,
"mean": [
0.485,
0.456,
0.406
],
"std": [
0.229,
0.224,
0.225
]
}