Package Detection using Yolo

This set of Notebooks was taken from https://github.com/nicknochnack/YOLO-Drowsiness-Detection by following the Youtube Tutorial from Nicholas Renotte's YouTube channel. The differences are the training dataset was modified to detect packages.

Getting started

Step 1. Clone this repository: https://github.com/nicknochnack/YOLO-Drowsiness-Detection

Step 2. Create a new virtual environment

python3 -m venv ypd

Step 3. Activate your virtual environment

source ypd/bin/activate # Linux

Step 4. Install dependencies and add virtual environment to the Python Kernel

python3 -m pip install --upgrade pip pip install ipykernel python3 -m ipykernel install --user --name=ypd

Step 5. Collect images using the Notebook 1. Image Collection.ipynb - ensure you change the kernel to the virtual environment ypd

Step 6. Manually divide collected images into two folders train and test. So now all folders and annotations should be split between the following two folders.
./Tensorflow/workspace/images/train
./Tensorflow/workspace/images/test

Step 7. Begin training process by opening Yolo Package Detection Tutorial.ipynb this notebook will walk you through installing Tensorflow Object Detection, making detections, saving and exporting your model.