diabolic6045 commited on
Commit
7a39f8c
1 Parent(s): 9d12963

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +27 -0
config.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "image-classification",
3
+ "architectures": ["VGG16"],
4
+ "finetuned": true,
5
+ "input_shape": [175, 175, 3],
6
+ "num_labels": 5,
7
+ "label2id": {
8
+ "Ahmedabad": 0,
9
+ "Delhi": 1,
10
+ "Kerala": 2,
11
+ "Kolkata": 3,
12
+ "Mumbai": 4
13
+ },
14
+ "id2label": {
15
+ "0": "Ahmedabad",
16
+ "1": "Delhi",
17
+ "2": "Kerala",
18
+ "3": "Kolkata",
19
+ "4": "Mumbai"
20
+ },
21
+ "pretrained_on": "ImageNet",
22
+ "fine_tuned_dataset": "Indian Cities",
23
+ "task": "image-classification",
24
+ "framework": "tensorflow",
25
+ "version": "1.0",
26
+ "description": "An image classification model specialized in identifying specific Indian cities from images."
27
+ }