mnist-net / README.md
Ritvik19's picture
Create README.md
90df5b0
|
raw
history blame
No virus
373 Bytes

Overview

MNIST Net V2 is a tensorflow Mobile Net V2 model fine tuned for 28x28x1 MNIST Handwritten Digits Classification The Images are internally rescaled in [0, 1] range and since Mobile Net V2 expects atleast 32x32 images, zero padding is added, also to translate the image to 3 color channels, we have used 1x1 Conv2D layer before the pretrained Mobile Net V2 layers