YOLO
15 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
YOLOV1Config1 methodsConfiguration for YOLOv1 (Redmon et al., CVPR 2016).
YOLOV1ForObjectDetection3 methodsYOLOv1 single-shot object detector (Redmon et al., CVPR 2016).
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).
YOLOV4Config1 methodsConfiguration for YOLOv4.
YOLOV4ForObjectDetection3 methodsYOLOv4 multi-scale object detector (Bochkovskiy et al., 2020).
Functions
yolo_v1→ YOLOV1ForObjectDetectionYOLOv1 with full Darknet backbone (Redmon et al., CVPR 2016).
yolo_v1_tiny→ YOLOV1ForObjectDetectionYOLOv1-Tiny with reduced Darknet backbone (Redmon et al., CVPR 2016).
yolo_v2→ YOLOV2ForObjectDetectionYOLOv2 / YOLO9000 with Darknet-19 backbone (Redmon & Farhadi, CVPR 2017).
yolo_v2_tiny→ YOLOV2ForObjectDetectionYOLOv2-Tiny — lightweight Darknet variant of YOLOv2.
yolo_v3→ YOLOV3ForObjectDetectionYOLOv3 with Darknet-53 backbone, 3-scale detection (Redmon & Farhadi, 2018).
yolo_v3_tiny→ _YOLOV3TinyYOLOv3-Tiny — 2-scale lightweight variant (Redmon & Farhadi, 2018).
yolo_v4→ YOLOV4ForObjectDetectionYOLOv4 — CSPDarknet-53 + SPP + PANet (Bochkovskiy et al., 2020).