Complete Guide to train YOLO

Train YOLO model for custom classes

This article contains a complete step-by-step tutorial to present an opportunity to retrain your YOLO model for custom classes, using Darkflow and Darknet frameworks.

Let's begin with generating a custom dataset for two classes, let them be Pooh Bear and Tiger from the Winnie-the-Pooh.

Step 1. Generate custom dataset;

Step 2. Recommended: install CUDA and cuDNN ;

Step 3. Train the model:

• using DarkNet;
• using DarkFlow;

Step 4. Convert DarkNet weights to protobuff file.