diabolic6045's picture
Create config.json
7a39f8c
raw
history blame
709 Bytes
{
"model_type": "image-classification",
"architectures": ["VGG16"],
"finetuned": true,
"input_shape": [175, 175, 3],
"num_labels": 5,
"label2id": {
"Ahmedabad": 0,
"Delhi": 1,
"Kerala": 2,
"Kolkata": 3,
"Mumbai": 4
},
"id2label": {
"0": "Ahmedabad",
"1": "Delhi",
"2": "Kerala",
"3": "Kolkata",
"4": "Mumbai"
},
"pretrained_on": "ImageNet",
"fine_tuned_dataset": "Indian Cities",
"task": "image-classification",
"framework": "tensorflow",
"version": "1.0",
"description": "An image classification model specialized in identifying specific Indian cities from images."
}