bus-classifier / deployment /model_metadata.json
Sathwik P
Initial deployment: Add Gradio app & model
829aece
raw
history blame contribute delete
752 Bytes
{
"class_names": [
"Alco-Brake_Device",
"Bus_-_Front_Side",
"First_Aid_Kit",
"Hat-rack_side-1",
"ITMS_Device_Functionality"
],
"class_to_idx": {
"Alco-Brake_Device": 0,
"Bus_-_Front_Side": 1,
"First_Aid_Kit": 2,
"Hat-rack_side-1": 3,
"ITMS_Device_Functionality": 4
},
"num_classes": 5,
"image_size": 224,
"model_name": "efficientnet_b0",
"validation_accuracy": 0.9922480583190918,
"training_samples": 1533,
"validation_samples": 387,
"normalization": {
"mean": [
0.485,
0.456,
0.406
],
"std": [
0.229,
0.224,
0.225
]
}
}