config.json

{
    "model_type": "resnet50",
    "input_size": [224, 224, 3],
    "num_classes": 6,
    "class_names": [
        "bath",
        "bed", 
        "din", 
        "kitchen", 
        "living"
    ]
}