What's On Pizza Project
What's On Pizza - project that gives you an ability to know pizza name by a photo. This repo contains server-side source code and similar training/validation scripts using MXNet and TensorFlow. You can compare two frameworks that solve one task. Client-side can be found here.
Build an ML model that would look at a picture of a pizza and output a list of possible pizza names. We trained a model for a single-label classification, where the output just the name of the pizza. Currently only ten pizza names.
Pizza images were collected with pizza scrapper (two python scripts for parsing pdf and html, examples of collected images can be found inside taxonomy folder). Classifier used for sorting pizza images and cleaning dataset from unrepresentative data. Trained models are used on the backend, which logic can be found here. It can be easily deployed with Docker.
We used Inception-BN model as a pre-trained model and fine-tune as a technique for changing initial model wights using collected data.
After training put output models at models folder otherwise, backend will not be able to work.
Please, write us for more details email@example.com