Model used in the project

This project uses one CNN model written on Pytorch. It was inspired by Pytorch challenge by Udacity and Facebook. Model was trained on ~6000 images, which were gathered thanks to the previous version of the project.

Convolutional neural net

The structure of model is simple:

FNN architecture

Code explanation