This repo let's you train a custom image detector using the state-of-the-art YOLOv3 computer vision algorithm. For a short write up check out this medium post. This repo works with TensorFlow 2.3 and Keras 2.4.
To build and test your YOLO object detection algorithm follow the below steps:
1_Image_Annotation
: Scripts and instructions on annotating images2_Training
: Scripts and instructions on training your YOLOv3 model3_Inference
: Scripts and instructions on testing your trained YOLO model on new images and videosData
: Input Data, Output Data, Model Weights and ResultsUtils
: Utility scripts used by main scriptsWith Google Colab you can skip most of the set up steps and start training your own model right away.
The only hard requirement is a running version of python 3.6 or 3.7. To install python 3.7 go to
and follow the installation instructions. Note that this repo has only been tested with python 3.6 and python 3.7 thus it is recommened to use either python3.6
or python3.7
.
To speed up training, it is recommended to use a GPU with CUDA support. For example on AWS you can use a p2.xlarge
instance (Tesla K80 GPU with 12GB memory). Inference speed on a typical CPU is approximately ~2 images per second. If you want to use your own machine, follow the instructions at tensorflow.org/install/gpu to install CUDA drivers. Make sure to install the correct version of CUDA and cuDNN.
Clone this repo with:
git clone https://github.com/AntonMu/TrainYourOwnYOLO
cd TrainYourOwnYOLO/
Create Virtual (Linux/Mac) Environment:
python3 -m venv env
source env/bin/activate
Make sure that, from now on, you run all commands from within your virtual environment.
Use the Github Desktop GUI to clone this repo to your local machine. Navigate to the TrainYourOwnYOLO
project folder and open a power shell window by pressing Shift + Right Click and selecting Open PowerShell window here
in the drop-down menu.
Create Virtual (Windows) Environment:
py -m venv env
.\env\Scripts\activate
Make sure that, from now on, you run all commands from within your virtual environment.
Install required packages (from within your virtual environment) via:
pip install -r requirements.txt
If this fails, you may have to upgrade your pip version first with pip install pip --upgrade
.
To test the cat face detector on test images located in TrainYourOwnYOLO/Data/Source_Images/Test_Images
run the Minimal_Example.py
script in the root folder with:
python Minimal_Example.py
The outputs are saved in TrainYourOwnYOLO/Data/Source_Images/Test_Image_Detection_Results
. This includes:
Detection_Results.csv
file with file names and locations of bounding boxes.
If you want to detect cat faces in your own pictures, replace the cat images in Data/Source_Images/Test_Images
with your own images.
To train your own custom YOLO object detector please follow the instructions detailed in the three numbered subfolders of this repo:
To make everything run smoothly it is highly recommended to keep the original folder structure of this repo!
Each *.py
script has various command line options that help tweak performance and change things such as input and output directories. All scripts are initialized with good default values that help accomplish all tasks as long as the original folder structure is preserved. To learn more about available command line options of a python script <script_name.py>
run:
python <script_name.py> -h
TrainYourOwnYOLO supports Weights & Biases to track your experiments online. Sign up at wandb.ai to get an API key and run:
wandb -login <API_KEY>
where <API_KEY>
is your Weights & Biases API key.
If you want to run multiple streams in parallel, head over to github.com/bertelschmitt/multistreamYOLO. Thanks to @bertelschmitt for putting the work into this.
Unless explicitly stated otherwise at the top of a file, all code is licensed under CC BY 4.0. This repo makes use of ilmonteux/logohunter which itself is inspired by qqwweee/keras-yolo3.
If you encounter any error, please make sure you follow the instructions exactly (word by word). Once you are familiar with the code, you're welcome to modify it as needed but in order to minimize error, I encourage you to not deviate from the instructions above. If you would like to file an issue, please use the provided template and make sure to fill out all fields.
If you encounter a FileNotFoundError
, Module not found
or similar error, make sure that you did not change the folder structure. Your directory structure must look exactly like this:
TrainYourOwnYOLO
ββββ 1_Image_Annotation
ββββ 2_Training
ββββ 3_Inference
ββββ Data
ββββ Utils
If you use a different name such as e.g. TrainYourOwnYOLO-master
you will have to specify the correct paths as command line arguments in every function call.
Don't use spaces in file or folder names, i.e. instead of my folder
use my_folder
.
If you are a Linux user and having trouble installing *.snap
package files try:
snap installβ--dangerous vott-2.1.0-linux.snap
See Snap Tutorial for more information.
If you have a newer version of python on your system, make sure that you create your virtual environment with version 3.7. You can use virtualenv for this:
pip install virtualenv
virtualenv env --python=python3.7
Then follow the same steps as above.
If you would like to file an issue, please use the provided issue template and make sure to complete all fields. This makes it easier to reproduce the issue for someone trying to help you.
Issues without a completed issue template will be closed and marked with the label "issue template not completed".
This work is licensed under a Creative Commons Attribution 4.0 International License. This means that you are free to:
Under the following terms:
Attribution
Cite as:
@misc{TrainYourOwnYOLO,
title = {TrainYourOwnYOLO: Building a Custom Object Detector from Scratch},
author = {Anton Muehlemann},
year = {2019},
url = {https://github.com/AntonMu/TrainYourOwnYOLO},
doi = {10.5281/zenodo.5112375}
}
If your work doesn't include a citation list, simply link this github repo!