A tutorial on training a DarkNet YOLOv4 model for the CrowdHuman dataset
YOLOv4 CrowdHuman Tutorial
This is a tutorial demonstrating how to train a YOLOv4 people detector using Darknet and the CrowdHuman dataset.
Setup
If you are going to train the model on Google Colab, you could skip this section and jump straight to Training on Google Colab.
Otherwise, to run training locally, you need to have a x86_64 PC with a decent GPU. For example, I mainly test the code in this repository using a desktop PC with:
- NVIDIA GeForce RTX 2080 Ti
- Ubuntu 18.04.5 LTS (x86_64)
In addition, you should have OpenCV (including python3 “cv2” module) installed properly on the local PC since both the data preparation code and “darknet” would require it.
Preparing training data
For training on the local PC, I use