File size: 1,660 Bytes
230507a bb333ba 230507a bb333ba d1dcff9 bb333ba 67c4d6d bb333ba 67c4d6d 230507a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
---
license: apache-2.0
datasets:
- mnist
language:
- en
metrics:
- accuracy
library_name: keras
---
# Description
This model is a convolutional neural network built with transfer learning using the pre-trained model 'VGG16.' The 'block5_conv1' layer is retrained, and a final dense layer with 128 neurons is added.
The model will be used as a preliminary step in solving Sudokus through linear programming. Model It is responsible for classifying the content of each sudoku cell:
- class_0: empty cell
- class_1: cell contains the number 1
- class_2: cell contains the number 2
- class_3: cell contains the number 3
- class_4: cell contains the number 4
- class_5: cell contains the number 5
- class_6: cell contains the number 6
- class_7: cell contains the number 7
- class_8: cell contains the number 8
- class_9: cell contains the number 9
The dataset is constructed with balanced classes using images from the famous "MNIST digits classification" dataset, as well as images of numbers written digitally.
# Dataset schema
The image size it is 28x28 pixels. After applying data augmentation to the dataset, the total number of images is as follows:
- Training images: 5,600
- Validation images: 2,400
- Test images: 2,000
Test Accuracy: 0.9810
# Other validations:
An initial validation is performed. It remains pending to increase the size of the validations to understand the reliability of the mode
<div style="text-align: center;">
<img src="https://i.imgur.com/kdj9udt.jpg" width="300">
</div>
</div>
The results of the inference are as follows:
<div style="text-align: center;">
<img src="https://i.imgur.com/U2MJzH6.jpg" width="500">
</div> |