/deepfood

Food image ranking using deep learning

Primary LanguageJupyter Notebook

Deepfood - Image Retrieval System in Production

Blog post on this project: https://yonigottesman.github.io/2020/05/15/deep-search-aws.html

System Arcitecture

Project Structure

Starlette web app

Web application that receives an image from user and return most similar images from index.

Training and Indexing Scripts

  • Notebook that fine tunes resnet34 pretrained model on food images.
  • Notebook that creates Annoy index.
  • Script that creates single folder with all images to index.
  • Notebbok to better understand embeddings.