BrickShapeRecognitionModel

A neural network model based on TensorFlow that predicts shape of brick

Problem definition

A model needs to recognize following brick shapes:

'bruk_prosty'
'master'
'metro'
'metro_xl'
'podwojne_t'
'trend'
'wena'

In order to do this I converted all brick images into grayscale and gave it to neural network.

Here are some sample images:

Model resulted in 99% accuracy on validation set and 100% on test set that had 30 images.