Scaled YOLOv4

Tue, 05 Jan 2021 00:00:00 +0000

Chien-Yao Wang, Alexey Bochkovskiy, and Hong-Yuan Mark Liao (more commonly known by their GitHub monikers, WongKinYiu and AlexyAB) have propelled the YOLOv4 model forward by efficiently scaling the network’s design and scale, surpassing the previous state-of-the-art EfficientDet published earlier this year by the Google Research/Brain team.

Train scaled YOLOv4 (PyTorch)

The Scaled-YOLOv4 implementation is written in the YOLOv5 PyTorch framework. Training scaled YOLOv4 is similar to training YOLOv5.

Here is t he Scaled-YOLOv4 repo, though you will notice that WongKinYiu has provided it there predominantly for research replication purposes and there are not many instructions for training on your own dataset. To train on your own data, our guide on training YOLOv5 in PyTorch on custom data will be useful, as it is a very similar training procedure.

Tutorials from Roboflow:

My Colab Notebook: yolov4_scaled.ipynb

Train scaled YOLOv4 (Darknet)

YOLOv4-csp training is also supported by Darknet. Training yolov4-csp is similar to training yolov4 and yolov4-tiny. Slight difference:

For config file, use yolov4-csp.cfg
For pretrained weights, use yolov4-csp.weights
For pretrained convolutional layer weights, use yolov4-csp.conv.142

Reference

Scaled YOLOv4 paper
Github repo: WongKinYiu/ScaledYOLOv4 (Different size of model in different branch)
Blog post from AlexAB: Scaled YOLO v4 is the best neural network for object detection on MS COCO dataset
Tutorials blog posts:

YOLOv3: Train on Custom Dataset

Tue, 05 Jan 2021 00:00:00 +0000

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

Scaled YOLOv4 | Haobin Tan

Scaled YOLOv4

Train scaled YOLOv4 (PyTorch)

Train scaled YOLOv4 (Darknet)

Reference

YOLOv3: Train on Custom Dataset

Reference