deployment/model_metadata.json

{
    "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
        ]
    }
}