giyomdabin commited on
Commit
02ca4e4
·
1 Parent(s): 4840c6c

chore: add config.json with model metadata for YOLOv8

Browse files
Files changed (1) hide show
  1. config.json +37 -0
config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "yolos",
3
+ "architecture": "YOLOv8",
4
+ "framework": "pytorch",
5
+ "num_labels": 13,
6
+ "id2label": {
7
+ "0": "AW",
8
+ "1": "BL2",
9
+ "2": "DCE",
10
+ "3": "GM",
11
+ "4": "HCS_HWPB",
12
+ "5": "HCS_HWPF",
13
+ "6": "HN",
14
+ "7": "HNT1B",
15
+ "8": "HNT1F",
16
+ "9": "MKFC",
17
+ "10": "N1KBB",
18
+ "11": "N1KBF",
19
+ "12": "PNX"
20
+ },
21
+ "label2id": {
22
+ "AW": 0,
23
+ "BL2": 1,
24
+ "DCE": 2,
25
+ "GM": 3,
26
+ "HCS_HWPB": 4,
27
+ "HCS_HWPF": 5,
28
+ "HN": 6,
29
+ "HNT1B": 7,
30
+ "HNT1F": 8,
31
+ "MKFC": 9,
32
+ "N1KBB": 10,
33
+ "N1KBF": 11,
34
+ "PNX": 12
35
+ }
36
+ }
37
+