From fc9a45ac615c3be25eca501ca6963a36c317447e Mon Sep 17 00:00:00 2001 From: Edmond Yoo <hj3yoo@uwaterloo.ca> Date: Fri, 07 Sep 2018 16:58:15 +0000 Subject: [PATCH] day1 update --- figures/1_augmented_set_example_3.jpg | 0 Makefile | 4 ++-- cfg/tiny_yolo.cfg | 2 +- darknet | 0 figures/1_augmented_set_example_4.jpg | 0 anchors.txt | 1 + figures/1_augmented_set_example_1.jpg | 0 README.md | 13 ++++++++++++- figures/1_augmented_set_example_2.jpg | 0 9 files changed, 16 insertions(+), 4 deletions(-) diff --git a/Makefile b/Makefile index b686b67..ef8b91e 100644 --- a/Makefile +++ b/Makefile @@ -1,7 +1,7 @@ GPU=1 -CUDNN=0 +CUDNN=1 CUDNN_HALF=0 -OPENCV=0 +OPENCV=1 AVX=0 OPENMP=0 LIBSO=0 diff --git a/README.md b/README.md index d7344a5..ca7085e 100644 --- a/README.md +++ b/README.md @@ -26,4 +26,15 @@ <img src="https://github.com/hj3yoo/darknet/blob/master/figures/0_detection_result_5.jpg" width="360"> <img src="https://github.com/hj3yoo/darknet/blob/master/figures/0_detection_result_6.jpg" width="360"> <img src="https://github.com/hj3yoo/darknet/blob/master/figures/0_detection_result_7.jpg" width="360"> -The second and third problems should easily be solved by further augmenting the dataset with random lighting and image skew. I'll have to think more about the first problem, though. \ No newline at end of file +The second and third problems should easily be solved by further augmenting the dataset with random lighting and image skew. I'll have to think more about the first problem, though. + +## Day 1 +----------------------- + +Added several image augmentation techniques to apply to the training set: noise, dropout, light variation, and glaring: + +<img src="https://github.com/hj3yoo/darknet/blob/master/figures/1_augmented_set_example_1.jpg" width="360"> <img src="https://github.com/hj3yoo/darknet/blob/master/figures/1_augmented_set_example_2.jpg" width="360"> <img src="https://github.com/hj3yoo/darknet/blob/master/figures/1_augmented_set_example_3.jpg" width="360"> <img src="https://github.com/hj3yoo/darknet/blob/master/figures/1_augmented_set_example_4.jpg" width="360"> + +Currently trying to generate enough images to start model training. Hopefully this helps. + +Recompiled darknet with OpenCV and CUDNN installed, and recalculated anchors. \ No newline at end of file diff --git a/anchors.txt b/anchors.txt new file mode 100644 index 0000000..5685246 --- /dev/null +++ b/anchors.txt @@ -0,0 +1 @@ +118.3429,137.0897, 95.8160,181.9724, 140.4955,166.7423, 112.7262,220.6808, 129.2741,198.9876, 159.0679,197.4912, 138.1861,243.0256, 167.4683,229.0091, 165.0264,255.0887 \ No newline at end of file diff --git a/cfg/tiny_yolo.cfg b/cfg/tiny_yolo.cfg index afc5564..468f33b 100644 --- a/cfg/tiny_yolo.cfg +++ b/cfg/tiny_yolo.cfg @@ -115,7 +115,7 @@ activation=linear [region] -anchors = 0.738768,0.874946, 2.42204,2.65704, 4.30971,7.04493, 10.246,4.59428, 12.6868,11.8741 +anchors = 118.3429,137.0897, 95.8160,181.9724, 140.4955,166.7423, 112.7262,220.6808, 129.2741,198.9876, 159.0679,197.4912, 138.1861,243.0256, 167.4683,229.0091, 165.0264,255.0887 bias_match=1 classes=1 coords=4 diff --git a/darknet b/darknet index 9d41f8a..b719fa2 100755 --- a/darknet +++ b/darknet Binary files differ diff --git a/figures/1_augmented_set_example_1.jpg b/figures/1_augmented_set_example_1.jpg new file mode 100644 index 0000000..1f75ac5 --- /dev/null +++ b/figures/1_augmented_set_example_1.jpg Binary files differ diff --git a/figures/1_augmented_set_example_2.jpg b/figures/1_augmented_set_example_2.jpg new file mode 100644 index 0000000..5eb1c4e --- /dev/null +++ b/figures/1_augmented_set_example_2.jpg Binary files differ diff --git a/figures/1_augmented_set_example_3.jpg b/figures/1_augmented_set_example_3.jpg new file mode 100644 index 0000000..9d41972 --- /dev/null +++ b/figures/1_augmented_set_example_3.jpg Binary files differ diff --git a/figures/1_augmented_set_example_4.jpg b/figures/1_augmented_set_example_4.jpg new file mode 100644 index 0000000..d4897ae --- /dev/null +++ b/figures/1_augmented_set_example_4.jpg Binary files differ -- Gitblit v1.10.0