You Only Look Once: Unified, Real-Time Object Detection
Top Cited Papers
- 1 June 2016
- conference paper
- conference paper
- Published by Institute of Electrical and Electronics Engineers (IEEE)
- p. 779-788
- https://doi.org/10.1109/cvpr.2016.91
Abstract
We present YOLO, a new approach to object detection. Prior work on object detection repurposes classifiers to perform detection. Instead, we frame object detection as a regression problem to spatially separated bounding boxes and associated class probabilities. A single neural network predicts bounding boxes and class probabilities directly from full images in one evaluation. Since the whole detection pipeline is a single network, it can be optimized end-to-end directly on detection performance. Our unified architecture is extremely fast. Our base YOLO model processes images in real-time at 45 frames per second. A smaller version of the network, Fast YOLO, processes an astounding 155 frames per second while still achieving double the mAP of other real-time detectors. Compared to state-of-the-art detection systems, YOLO makes more localization errors but is less likely to predict false positives on background. Finally, YOLO learns very general representations of objects. It outperforms other detection methods, including DPM and R-CNN, when generalizing from natural images to other domains like artwork.Keywords
Other Versions
This publication has 20 references indexed in Scilit:
- ImageNet Large Scale Visual Recognition ChallengeInternational Journal of Computer Vision, 2015
- The Pascal Visual Object Classes Challenge: A RetrospectiveInternational Journal of Computer Vision, 2014
- Scalable Object Detection Using Deep Neural NetworksPublished by Institute of Electrical and Electronics Engineers (IEEE) ,2014
- Rich Feature Hierarchies for Accurate Object Detection and Semantic SegmentationPublished by Institute of Electrical and Electronics Engineers (IEEE) ,2014
- Fast, Accurate Detection of 100,000 Object Classes on a Single MachinePublished by Institute of Electrical and Electronics Engineers (IEEE) ,2013
- Selective Search for Object RecognitionInternational Journal of Computer Vision, 2013
- Diagnosing Error in Object DetectorsLecture Notes in Computer Science, 2012
- Learning to Localize Objects with Structured Output RegressionLecture Notes in Computer Science, 2008
- Robust Real-Time Face DetectionInternational Journal of Computer Vision, 2004
- An extended set of Haar-like features for rapid object detectionPublished by Institute of Electrical and Electronics Engineers (IEEE) ,2003