/image-recog

Image recognition using tensorflow.

Primary LanguageJupyter Notebook

Computer Vision Template

Setup

Use docker compose up to use the project.

Poetry is not supported rn, so add additional reqs to requirements.txt.

Project Overview

This project is based on this research paper but instead of random forests, we use convolutional neural networks, abstracted via the TensorFlow Keras API.

Of the dataset this project only uses the pizza and steak parts, with a 80-20 split for training and validation.

Folder Structure

Your folder structure should look like this when the project is setup after you download the dataset. Where pizza and steak are folders copied from the images directory.

Folder Structure