dmedhi commited on
Commit
8443f92
·
verified ·
1 Parent(s): ae7445b

Add model files and configurations

Browse files
Files changed (1) hide show
  1. config.json +27 -0
config.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "resnet18-garbage-classifier",
3
+ "model_type": "resnet",
4
+ "architecture": "resnet18",
5
+ "task": "image-classification",
6
+ "num_labels": 8,
7
+ "id2label": {
8
+ "0": "Garbage",
9
+ "1": "Cardboard",
10
+ "2": "Garbage",
11
+ "3": "Glass",
12
+ "4": "Metal",
13
+ "5": "Paper",
14
+ "6": "Plastic",
15
+ "7": "Trash"
16
+ },
17
+ "label2id": {
18
+ "Garbage": 0,
19
+ "Cardboard": 1,
20
+ "Glass": 3,
21
+ "Metal": 4,
22
+ "Paper": 5,
23
+ "Plastic": 6,
24
+ "Trash": 7
25
+ },
26
+ "license": "apache-2.0"
27
+ }