| New file |
| | |
|---|
| | | |
|---|
| | | # 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. |
|---|
| | | |
|---|
| | | ## 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. |
|---|
| | | |
|---|
| | | ----------------------- |
|---|
| | | |
|---|
| | | I've ran a quick training with tiny_yolo configuration with new training data, and Voila! The model performs significantly better than the last iteration, even against some hard images with glaring & skew! The first prediction model can't detect anything from these new test images, so this is a huge improvement to the model :) |
|---|
| | | |
|---|
| | | <img src="https://github.com/hj3yoo/darknet/blob/master/figures/1_detection_result_1.jpg" width="360"> <img src="https://github.com/hj3yoo/darknet/blob/master/figures/1_detection_result_2.jpg" width="360"> <img src="https://github.com/hj3yoo/darknet/blob/master/figures/1_detection_result_3.jpg" width="360"> <img src="https://github.com/hj3yoo/darknet/blob/master/figures/1_detection_result_4.jpg" width="360"> <img src="https://github.com/hj3yoo/darknet/blob/master/figures/1_detection_result_5.jpg" width="360"> <img src="https://github.com/hj3yoo/darknet/blob/master/figures/1_detection_result_6.jpg" width="360"> |
|---|
| | | |
|---|
| | | <img src="https://github.com/hj3yoo/darknet/blob/master/figures/1_learning_curve.jpg" width="360"> |
|---|
