config.json

{
    "model_name": "resnet50",
    "num_classes": 10,
    "input_size": 224,
    "model_type" : "ResNet",
    "classes": [
        "battery",
        "biological",
        "cardboard",
        "clothes",
        "glass",
        "metal",
        "paper",
        "plastic",
        "shoes",
        "trash"
    ],
    "normalize_mean": [
        0.485,
        0.456,
        0.406
    ],
    "normalize_std": [
        0.229,
        0.224,
        0.225
    ],
    "version": "1.0.0",
    "description": "ResNet50 model fine-tuned for garbage classification"
}