flerokoo / tensorflow-pixel-food

Another ML learning project

Another Machine Learning Learning Project

Same as Pixel Food Classificator, but, unlike it, was done with Keras module of Tensorflow.

Model structure

• Two convolutional layers with 3x3 kernel and 64 filters and ReLU activation function
• Dense layer with 64 neurons and ReLU
• Dense layer with 3 neurons and softmax activation

Result

Some plots from Tensorboard:

Usage

Convert images to input format of shape (N_images, 32, 32, 1):

python prepare_data.py

Train model:

python train_model.py

Test model:

python test_model.py

Check tensorboard:

tensorboard --logdir=logs/