-
Notifications
You must be signed in to change notification settings - Fork 10
/
config.json
37 lines (33 loc) · 734 Bytes
/
config.json
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
{
"env":{
"gpu":"0,1,2,3"
},
"model":{
"architecture":"Tiny Yolo",
"input_size":416,
"anchors":[0.57273, 0.677385, 1.87446, 2.06253, 3.33843, 5.47434, 7.88282, 3.52778, 9.77052, 9.16828],
"max_box_per_image":10,
"labels":["RBC"]
},
"train":{
"train_image_folder":"RBC_datasets/JPEGImages/",
"train_annot_folder":"RBC_datasets/Annotations/",
"train_times":5,
"pretrained_weights":"",
"batch_size":1,
"learning_rate":1e-4,
"nb_epoch":50,
"warmup_batches":250,
"object_scale":5.0,
"no_object_scale":1.0,
"coord_scale":1.0,
"class_scale":1.0,
"saved_weights_name":"models/tiny_yolo.h5",
"debug":0
},
"valid":{
"valid_image_folder":"",
"valid_annot_folder":"",
"valid_times":1
}
}