YOLOv3: Train on Custom Dataset

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

Reference