RaphaelKalandadze commited on
Commit
833b248
·
1 Parent(s): ffa8a1f

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +12 -7
config.json CHANGED
@@ -1,10 +1,8 @@
1
  {
2
- "task": "detect",
3
- "mode": "predict",
4
- "conf": 0.5,
5
- "imgsz": 640,
6
- "nc": 17,
7
- "model_type": "yolos",
8
  "id2label": {
9
  "0":"aop",
10
  "1":"asc",
@@ -24,6 +22,10 @@
24
  "15":"vegetarian",
25
  "16":"vpf"
26
  },
 
 
 
 
27
  "label2id": {
28
  "aop":0,
29
  "asc":1,
@@ -42,5 +44,8 @@
42
  "vbf":14,
43
  "vegetarian":15,
44
  "vpf":16
45
- }
 
 
 
46
  }
 
1
  {
2
+ "_name_or_path": "RaphaelKalandadze/tmp_trainer",
3
+ "architectures": [
4
+ "YolosForObjectDetection"
5
+ ],
 
 
6
  "id2label": {
7
  "0":"aop",
8
  "1":"asc",
 
22
  "15":"vegetarian",
23
  "16":"vpf"
24
  },
25
+ "image_size": [
26
+ 640,
27
+ 640
28
+ ],
29
  "label2id": {
30
  "aop":0,
31
  "asc":1,
 
44
  "vbf":14,
45
  "vegetarian":15,
46
  "vpf":16
47
+ },
48
+ "model_type": "yolos",
49
+ "num_channels": 3,
50
+ "transformers_version": "4.21.1",
51
  }