diff --git a/README.md b/README.md index 31249c1..70af810 100755 --- a/README.md +++ b/README.md @@ -191,6 +191,8 @@ All of weights of this trained model grab from [Drive_Weights](https://drive.goo | YOLO3 Soiling | [weight](https://drive.google.com/open?id=1YLgkn1wL5xAGOpwd2gzdfsJVGYPzszn-) | [config](config_full_yolo_fault_1_infer.json) | | YOLO3 Diode | [weight](https://drive.google.com/open?id=1VUtrK9JVTbzBw5dX7_dgLTMToFHbAJl1) | [config](config_full_yolo_fault_4_infer.json) | +| YOLO3 Affected Cell | [weight_lack](...) | [config](config_full_yolo_fault_2_infer.json) | + ## Panel Detector ### SDD7 On folder [Result ssd7 panel](Result_ssd7_panel/) show code (jupyter notebook), weight and result of this model (mAP 89.8%). diff --git a/config_full_yolo_fault_2_infer.json b/config_full_yolo_fault_2_infer.json new file mode 100755 index 0000000..e6712f3 --- /dev/null +++ b/config_full_yolo_fault_2_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": ["1"], + "backend": "keras-yolo3-master/full_yolo_backend.h5" + }, + + "train": { + "train_image_folder": "Train&Test_H/Train/images/", + "train_annot_folder": "Train&Test_H/Train/anns/", + "cache_name": "Result_yolo3_fault_2/experimento_fault_2_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_2", + "saved_weights_name": "Result_yolo3_fault_2/yolo3_full_fault_2.h5", + "debug": true + }, + + "valid": { + "valid_image_folder": "Train&Test_H/Test/images/", + "valid_annot_folder": "Train&Test_H/Test/anns/", + "cache_name": "Result_yolo3_fault_2/val_fault_2.pkl", + + "valid_times": 1 + }, + "test": { + "test_image_folder": "Train&Test_H/Test/images/", + "test_annot_folder": "Train&Test_H/Test/anns/", + "cache_name": "Result_yolo3_fault_2/test_fault_2.pkl", + + "test_times": 1 + } +} diff --git a/config_full_yolo_fault_2_train.json b/config_full_yolo_fault_2_train.json new file mode 100755 index 0000000..e6712f3 --- /dev/null +++ b/config_full_yolo_fault_2_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": "keras-yolo3-master/full_yolo_backend.h5" + }, + + "train": { + "train_image_folder": "Train&Test_H/Train/images/", + "train_annot_folder": "Train&Test_H/Train/anns/", + "cache_name": "Result_yolo3_fault_2/experimento_fault_2_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_2", + "saved_weights_name": "Result_yolo3_fault_2/yolo3_full_fault_2.h5", + "debug": true + }, + + "valid": { + "valid_image_folder": "Train&Test_H/Test/images/", + "valid_annot_folder": "Train&Test_H/Test/anns/", + "cache_name": "Result_yolo3_fault_2/val_fault_2.pkl", + + "valid_times": 1 + }, + "test": { + "test_image_folder": "Train&Test_H/Test/images/", + "test_annot_folder": "Train&Test_H/Test/anns/", + "cache_name": "Result_yolo3_fault_2/test_fault_2.pkl", + + "test_times": 1 + } +}