Create config.json
Browse files- config.json +53 -0
config.json
ADDED
@@ -0,0 +1,53 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"architectures": [
|
3 |
+
"GGHLForObjectDetection"
|
4 |
+
],
|
5 |
+
"model_type": "gghl",
|
6 |
+
"task": "object-detection",
|
7 |
+
"id2label": {
|
8 |
+
"0": "plane",
|
9 |
+
"1": "baseball-diamond",
|
10 |
+
"2": "bridge",
|
11 |
+
"3": "ground-track-field",
|
12 |
+
"4": "small-vehicle",
|
13 |
+
"5": "large-vehicle",
|
14 |
+
"6": "ship",
|
15 |
+
"7": "tennis-court",
|
16 |
+
"8": "basketball-court",
|
17 |
+
"9": "storage-tank",
|
18 |
+
"10": "soccer-ball-field",
|
19 |
+
"11": "roundabout",
|
20 |
+
"12": "harbor",
|
21 |
+
"13": "swimming-pool",
|
22 |
+
"14": "helicopter"
|
23 |
+
},
|
24 |
+
"label2id": {
|
25 |
+
"plane": "0",
|
26 |
+
"baseball-diamond": "1",
|
27 |
+
"bridge": "2",
|
28 |
+
"ground-track-field": "3",
|
29 |
+
"small-vehicle": "4",
|
30 |
+
"large-vehicle": "5",
|
31 |
+
"ship": "6",
|
32 |
+
"tennis-court": "7",
|
33 |
+
"basketball-court": "8",
|
34 |
+
"storage-tank": "9",
|
35 |
+
"soccer-ball-field": "10",
|
36 |
+
"roundabout": "11",
|
37 |
+
"harbor": "12",
|
38 |
+
"swimming-pool": "13",
|
39 |
+
"helicopter": "14"
|
40 |
+
},
|
41 |
+
"num_classes": 15,
|
42 |
+
"input_size": 800,
|
43 |
+
"strides": [
|
44 |
+
8,
|
45 |
+
16,
|
46 |
+
32
|
47 |
+
],
|
48 |
+
"confidence_threshold": 0.2,
|
49 |
+
"nms_threshold": 0.45,
|
50 |
+
"iou_threshold": 0.5,
|
51 |
+
"max_detections": 100,
|
52 |
+
"transformers_version": "4.28.0"
|
53 |
+
}
|