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 --- README.md | 42 ++++++++++++++++++++++++++++++++++++++---- 1 files changed, 38 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index 7118815..ca7085e 100644 --- a/README.md +++ b/README.md @@ -1,6 +1,40 @@ - -#Darknet# -Darknet is an open source neural network framework written in C and CUDA. It is fast, easy to install, and supports CPU and GPU computation. +# Magic: The Gathering Card Detection Model -For more information see the [Darknet project website](http://pjreddie.com/darknet). +This is a fork of [Yolo-v3 and Yolo-v2 for Windows and Linux by AlexeyAB](https://github.com/AlexeyAB/darknet#how-to-compile-on-linux) for creating a custom model for [My MTG card detection project](https://github.com/hj3yoo/MTGCardDetector). + +## Day ~0: Sep 6th, 2018 +--------------------- + +Uploading all the progresses on the model training for the last few days. + +First batch of model training is completed, where I used ~40,000 generated images of MTG cards laid out in one of the pre-defined pattern. + +<img src="https://github.com/hj3yoo/darknet/blob/master/figures/0_training_set_example_1.jpg" width="360"> <img src="https://github.com/hj3yoo/darknet/blob/master/figures/0_training_set_example_2.jpg" width="360"> <img src="https://github.com/hj3yoo/darknet/blob/master/figures/0_training_set_example_3.jpg" width="360"> + +After 5000 training epochs, the model got 88% validation accuracy on the generated test set. + +<img src="https://github.com/hj3yoo/darknet/blob/master/figures/0_detection_result_1.jpg" width="360"> <img src="https://github.com/hj3yoo/darknet/blob/master/figures/0_detection_result_2.jpg" width="360"> <img src="https://github.com/hj3yoo/darknet/blob/master/figures/0_detection_result_3.jpg" width="360"> <img src="https://github.com/hj3yoo/darknet/blob/master/figures/0_detection_result_4.jpg" width="360"> + +However, there are some blind spots on the model, notably: + +- Fails to spot some of the obscured cards, where only a fraction of them are shown. +- Fairly fragile against any glaring or light variations. +- Cannot detect any skewed cards. + +Example of bad detections: + +<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. + +## 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 -- Gitblit v1.10.0