filipsedivy commited on
Commit
0479e7c
1 Parent(s): 6da988a

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -1
config.json CHANGED
@@ -1,5 +1,7 @@
1
  {
2
  "labels": ["Nature", "Other", "Person", "Person half naked", "Person naked"],
3
  "id2label": {"0": "Nature", "1": "Other", "2": "Person", "3": "Person half naked", "4": "Person naked"},
4
- "input_size": [3, 224, 224]
 
 
5
  }
 
1
  {
2
  "labels": ["Nature", "Other", "Person", "Person half naked", "Person naked"],
3
  "id2label": {"0": "Nature", "1": "Other", "2": "Person", "3": "Person half naked", "4": "Person naked"},
4
+ "label2id": {"Nature": "0", "Other": "1", "Person": "2", "Person half naked": "3", "Person naked": "4"},
5
+ "image_size": 224,
6
+ "num_channels": 3
7
  }