50 lines
1.6 KiB
JSON
Executable File
50 lines
1.6 KiB
JSON
Executable File
{
|
|
"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": ["4"],
|
|
"backend": "keras-yolo3-master/full_yolo_backend.h5"
|
|
},
|
|
|
|
"train": {
|
|
"train_image_folder": "Train&Test_D/Train/images/",
|
|
"train_annot_folder": "Train&Test_D/Train/anns/",
|
|
"cache_name": "Result_yolo3_fault_4/Result_yolo3_fault_4.pkl",
|
|
|
|
"train_times": 1,
|
|
|
|
"batch_size": 2,
|
|
"learning_rate": 1e-4,
|
|
"nb_epochs": 500,
|
|
"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_4/yolo3_full_fault_4.h5",
|
|
"debug": true
|
|
},
|
|
|
|
"valid": {
|
|
"valid_image_folder": "Train&Test_D/Test/images/",
|
|
"valid_annot_folder": "Train&Test_D/Test/anns/",
|
|
"cache_name": "Result_yolo3_fault_4/Result_yolo3_fault_4.pkl",
|
|
|
|
"valid_times": 1
|
|
},
|
|
"test": {
|
|
"test_image_folder": "../Train&Test_D/Test/images/",
|
|
"test_annot_folder": "../Train&Test_D/Test/anns/",
|
|
"cache_name": "../Result_yolo3_fault_4/Result_yolo3_fault_4.pkl",
|
|
|
|
"test_times": 1
|
|
}
|
|
}
|