Hsueh1001 commited on
Commit
41fbe4d
1 Parent(s): 77b437f

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +19 -0
config.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "YolosForObjectDetection"
4
+ ],
5
+ "model_type": "yolos",
6
+ "input_size": [640, 640],
7
+ "anchors": [
8
+ [10, 13],
9
+ [16, 30],
10
+ [33, 23]
11
+ ],
12
+ "classes": ["0", "1"],
13
+ "output_layers": [
14
+ {"layer_size": [52, 52], "anchors": 3},
15
+ {"layer_size": [26, 26], "anchors": 3},
16
+ {"layer_size": [13, 13], "anchors": 3}
17
+ ],
18
+ "nms_threshold": 0.5
19
+ }