YOLOv3: Train on Custom Dataset

Training YOLOv3 as well as YOLOv3 tiny on custom dataset is similar to training YOLOv4 and YOLOv4 tiny. Only some steps need to be adjusted for YOLOv3 and YOLOv3 tiny:

• In step 1, we create our custom config file based on cfg/yolov3.cfg (YOLOv3) and cfg/yolov3-tiny.cfg (YOLOv3 tiny). Then adjust batch, subdivisions, steps, width, height, classes, and filters just as for YOLOv4.
• In step 6, download different pretrained weights for the convolutional layers
  • for yolov3.cfg, yolov3-spp.cfg (154 MB): darknet53.conv.74
  • for yolov3-tiny-prn.cfg , yolov3-tiny.cfg (6 MB): yolov3-tiny.conv.11

Reference

• Tutorial from darknet repo: How to train (to detect your custom objects)

• How to train YOLOv3 on the custom dataset