How to train Yolov3 and Yolov5 with your own datasets

Introduction

Set up Yolov5 environment

git clone https://github.com/ultralytics/yolov5
cd yolov5
pip install -r requirements.txt

Label images

There many object labelling tools, such as labelImg, CVAT and VoTT. Here we will use VoTT to label our images. Note labelImg is easy to use if you only label one class of objects.

Config Yolov5 network

Train Yolov5 model

Inference with model