/yolo

working knowledge of yolo algorithm

Primary LanguagePythonMIT LicenseMIT

Yolo for Culinary

API Documentation

empire-penguin.github.io/yolo

Build Documentation

cd docs
make html

Installation Instructions

Clone Repo

git clone https://github.com/empire-penguin/yolo
cd yolo

Create a new virtual environment

python3 -m venv venv

Activate the virtual environment

  • On Windows:

    venv\Scripts\activate

  • On macOS and Linux:

    source venv/bin/activate

Install requirements

pip3 install -r requirements.txt

Install pytorch

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

Install yolo

pip3 install .

Getting dataset

wget -P dataset http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar 

tar -xvf VOCtrainval_11-May-2012.tar

Usage

Run unit tests

python3 -m unittest

this will download the VOCdevkit used by yolo to train and find the features for the decision making process edict.py

Todo

Create / find a dataset to do transfer learning of grocery items