-
Notifications
You must be signed in to change notification settings - Fork 19
/
config_full_yolo_fault_1_train.json
executable file
·49 lines (41 loc) · 1.57 KB
/
config_full_yolo_fault_1_train.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
38
39
40
41
42
43
44
45
46
47
48
49
{
"model" : {
"min_input_size": 400,
"max_input_size": 400,
"anchors": [5,7, 10,14, 15, 15, 26,32, 45,119, 54,18, 94,59, 109,183, 200,21],
"labels": ["1"],
"backend": "keras-yolo3-master/full_yolo_backend.h5"
},
"train": {
"train_image_folder": "Train&Test_S/Train/images/",
"train_annot_folder": "Train&Test_S/Train/anns/",
"cache_name": "Result_yolo3_fault_1/experimento_fault_1_gpu.pkl",
"train_times": 1,
"batch_size": 2,
"learning_rate": 1e-4,
"nb_epochs": 200,
"warmup_epochs": 15,
"ignore_thresh": 0.5,
"gpus": "0,1",
"grid_scales": [1,1,1],
"obj_scale": 5,
"noobj_scale": 1,
"xywh_scale": 1,
"class_scale": 1,
"tensorboard_dir": "log_experimento_fault_gpu",
"saved_weights_name": "Result_yolo3_fault_1/yolo3_full_fault_1.h5",
"debug": true
},
"valid": {
"valid_image_folder": "Train&Test_S/Test/images/",
"valid_annot_folder": "Train&Test_S/Test/anns/",
"cache_name": "Result_yolo3_fault_1/val_fault_1.pkl",
"valid_times": 1
},
"test": {
"test_image_folder": "Train&Test_S/Test/images/",
"test_annot_folder": "Train&Test_S/Test/anns/",
"cache_name": "Result_yolo3_fault_1/test_fault_1.pkl",
"test_times": 1
}
}