From 01db677661e1797fa4f2b2c9d3c2e16cfb167017 Mon Sep 17 00:00:00 2001 From: Daniel Saavedra Date: Thu, 20 Feb 2020 13:55:34 -0300 Subject: [PATCH] change config --- ...son => config_full_yolo_fault_1_infer.json | 2 +- config_full_yolo_fault_1_train.json | 49 +++++++++++++++++++ config_full_yolo_fault_4_infer.json | 49 +++++++++++++++++++ ...son => config_full_yolo_fault_4_train.json | 0 4 files changed, 99 insertions(+), 1 deletion(-) rename config_full_yolo_fault_1.json => config_full_yolo_fault_1_infer.json (94%) create mode 100755 config_full_yolo_fault_1_train.json create mode 100755 config_full_yolo_fault_4_infer.json rename config_full_yolo_fault_4.json => config_full_yolo_fault_4_train.json (100%) diff --git a/config_full_yolo_fault_1.json b/config_full_yolo_fault_1_infer.json similarity index 94% rename from config_full_yolo_fault_1.json rename to config_full_yolo_fault_1_infer.json index 82d6bbb..7763f45 100755 --- a/config_full_yolo_fault_1.json +++ b/config_full_yolo_fault_1_infer.json @@ -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 }, diff --git a/config_full_yolo_fault_1_train.json b/config_full_yolo_fault_1_train.json new file mode 100755 index 0000000..6deb86a --- /dev/null +++ b/config_full_yolo_fault_1_train.json @@ -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 + } +} diff --git a/config_full_yolo_fault_4_infer.json b/config_full_yolo_fault_4_infer.json new file mode 100755 index 0000000..5c1d36f --- /dev/null +++ b/config_full_yolo_fault_4_infer.json @@ -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 + } +} diff --git a/config_full_yolo_fault_4.json b/config_full_yolo_fault_4_train.json similarity index 100% rename from config_full_yolo_fault_4.json rename to config_full_yolo_fault_4_train.json