change config
This commit is contained in:
@@ -28,7 +28,7 @@
|
||||
"class_scale": 1,
|
||||
|
||||
"tensorboard_dir": "log_experimento_fault_gpu",
|
||||
"saved_weights_name": "../Result_yolo3_fault_1/experimento_yolo3_full_fault.h5",
|
||||
"saved_weights_name": "Result_yolo3_fault_1/yolo3_full_fault_1.h5",
|
||||
"debug": true
|
||||
},
|
||||
|
||||
49
config_full_yolo_fault_1_train.json
Executable file
49
config_full_yolo_fault_1_train.json
Executable file
@@ -0,0 +1,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": "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
|
||||
}
|
||||
}
|
||||
49
config_full_yolo_fault_4_infer.json
Executable file
49
config_full_yolo_fault_4_infer.json
Executable file
@@ -0,0 +1,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": ["4"],
|
||||
"backend": "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
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user