mhassanch commited on
Commit
c389631
·
verified ·
1 Parent(s): 4430d8d

Create config.json

Browse files
Files changed (1) hide show
  1. 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
+ }