From c8b7dc4f215b89f821b3f242ccb2ada7f535efeb Mon Sep 17 00:00:00 2001 From: Edmond Yoo <hj3yoo@uwaterloo.ca> Date: Thu, 06 Sep 2018 17:26:56 +0000 Subject: [PATCH] readme day 0 --- README.md | 25 +++++++++++++++++++++++++ 1 files changed, 25 insertions(+), 0 deletions(-) diff --git a/README.md b/README.md index 8d0c8a0..d7344a5 100644 --- a/README.md +++ b/README.md @@ -1,4 +1,29 @@ + # Magic: The Gathering Card Detection Model 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. \ No newline at end of file -- Gitblit v1.10.0