| { | |
| "model_type": "image-classification", | |
| "framework": "tensorflow", | |
| "architecture": "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", | |
| "description": "An image classification model specialized in identifying specific Indian cities from images. Built using TensorFlow and saved in the SavedModel format.", | |
| "model_format": "SavedModel", | |
| "version": "1.0" | |
| } | |