Skip to content

Baseline model weights files

Compare
Choose a tag to compare
@GKalliatakis GKalliatakis released this 12 Jul 13:28
· 33 commits to master since this release

Add weights files for the baseline (trained from scratch) HRA-CNN model

_________________________________________________________________
Layer (type)                 Output Shape              Param #
=================================================================
input_1 (InputLayer)         (None, 224, 224, 3)       0
_________________________________________________________________
block1_conv1 (Conv2D)        (None, 224, 224, 32)      896
_________________________________________________________________
block1_pool (MaxPooling2D)   (None, 112, 112, 32)      0
_________________________________________________________________
block2_conv1 (Conv2D)        (None, 112, 112, 32)      9248
_________________________________________________________________
block2_pool (MaxPooling2D)   (None, 56, 56, 32)        0
_________________________________________________________________
block3_conv1 (Conv2D)        (None, 56, 56, 64)        18496
_________________________________________________________________
block3_pool (MaxPooling2D)   (None, 28, 28, 64)        0
_________________________________________________________________
flatten (Flatten)            (None, 50176)             0
_________________________________________________________________
fc1 (Dense)                  (None, 64)                3211328
_________________________________________________________________
dropout_1 (Dropout)          (None, 64)                0
_________________________________________________________________
predictions (Dense)          (None, 9)                 585
=================================================================
Total params: 3,240,553
Trainable params: 3,240,553
Non-trainable params: 0
_________________________________________________________________