YOLO
20 memberslucid.models.vision.yoloYOLO family — You Only Look Once object detectors.
Notes
- YOLOv1: Redmon et al., CVPR 2016
- YOLOv2: Redmon & Farhadi, CVPR 2017 (YOLO9000)
- YOLOv3: Redmon & Farhadi, arXiv 2018
- YOLOv4: Bochkovskiy et al., arXiv 2020
Redmon, Joseph, et al. "You Only Look Once: Unified, Real-Time Object Detection." Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 2016, pp. 779–788.
YOLOv1 reframes detection as a single regression problem. The input image is divided into an grid; each cell is responsible for objects whose centre falls inside it. For each cell the network predicts candidate boxes plus one shared class distribution over categories, so the head output has shape .
A single CNN (the Darknet trunk in the paper) processes the full image once and emits all detections in one forward pass. The confidence target is so that confidence encodes both objectness and localisation quality. Training uses a weighted sum-squared-error loss
where make the regression less sensitive to absolute scale and rebalance the dense background term.
Because the entire pipeline is a single CNN, YOLOv1 runs at 45 fps (155 fps for the tiny variant) — orders of magnitude faster than the two-stage R-CNN family — establishing the one-stage detection paradigm. Trade-off: each cell predicts a fixed small , limiting recall on clustered small objects, which is exactly what YOLOv2/v3/v4 progressively fix with anchors, multi-scale heads, and a stronger backbone/neck.
Classes
YOLOConfig1 methodsConfiguration for YOLOv1 (Redmon et al., CVPR 2016).
YOLOForObjectDetection3 methodsYOLOv1 single-shot object detector (Redmon et al., CVPR 2016).
MultiScaleResolution2 methodsThe input-size schedule of YOLOv2's multi-scale training (§2).
YOLOV2Config1 methodsConfiguration for YOLOv2 / YOLO9000 (Redmon & Farhadi, CVPR 2017).
YOLOV2ForObjectDetection3 methodsYOLOv2 / YOLO9000 object detector (Redmon & Farhadi, CVPR 2017).
YOLOV3Config1 methodsConfiguration for YOLOv3.
YOLOV3ForObjectDetection3 methodsYOLOv3 multi-scale object detector (Redmon & Farhadi, 2018).
YOLOV3TinyForObjectDetection3 methodsYOLOv3-Tiny — the 2-scale lightweight YOLOv3 (Redmon & Farhadi, 2018).
YOLOV4Config1 methodsConfiguration for YOLOv4.
YOLOV4ForObjectDetection3 methodsYOLOv4 multi-scale object detector (Bochkovskiy et al., 2020).
Functions
yolo→ YOLOForObjectDetectionYOLOv1 with full Darknet backbone (Redmon et al., CVPR 2016).
yolo_tiny→ YOLOForObjectDetectionYOLOv1-Tiny with reduced Darknet backbone (Redmon et al., CVPR 2016).
yolo_v2→ YOLOV2ForObjectDetection51MYOLOv2 / YOLO9000 with Darknet-19 backbone (Redmon & Farhadi, CVPR 2017).
yolo_v3→ YOLOV3ForObjectDetection61.9MYOLOv3 with Darknet-53 backbone, 3-scale detection (Redmon & Farhadi, 2018).
yolo_v3_tiny→ YOLOV3TinyForObjectDetection8.9MYOLOv3-Tiny — 2-scale lightweight variant (Redmon & Farhadi, 2018).
yolo_v4→ YOLOV4ForObjectDetection64.4MYOLOv4 — CSPDarknet-53 + SPP + PANet (Bochkovskiy et al., 2020).