Skip to content

2019.08 Release with CNN model

Latest
Compare
Choose a tag to compare
@MartinKondor MartinKondor released this 30 Jun 15:14
· 2 commits to master since this release

The model's summary:

_________________________________________________________________
Layer (type)                 Output Shape              Param #
=================================================================
conv2d_1 (Conv2D)            (None, 78, 88, 32)        896
_________________________________________________________________
conv2d_2 (Conv2D)            (None, 76, 86, 32)        9248
_________________________________________________________________
max_pooling2d_1 (MaxPooling2 (None, 38, 43, 32)        0
_________________________________________________________________
dropout_1 (Dropout)          (None, 38, 43, 32)        0
_________________________________________________________________
flatten_1 (Flatten)          (None, 52288)             0
_________________________________________________________________
dense_1 (Dense)              (None, 128)               6692992
_________________________________________________________________
dropout_2 (Dropout)          (None, 128)               0
_________________________________________________________________
dense_2 (Dense)              (None, 2)                 258
=================================================================
Total params: 6,703,394
Trainable params: 6,703,394
Non-trainable params: 0
_________________________________________________________________

Accuracy on train set: 99.9
Accuracy on test set: 88.99