你只需看一眼:统一、实时的目标检测

You Only Look Once: Unified, Real-Time Object Detection

约瑟夫·雷德蒙 Joseph Redmon · U. Washington · 2015-06-08 · arXiv:1506.02640 ↗ · 被引 46352

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

我们提出了 YOLO,一种新的目标检测方法。以往的目标检测工作将分类器重新用于检测。相反,我们将目标检测视为一个回归问题,预测空间分离的边界框和相关的类别概率。单个神经网络在一次评估中直接从完整图像中预测边界框和类别概率。由于整个检测管道是一个单一网络,它可以直接在检测性能上进行端到端优化。我们的统一架构非常快。我们的基础 YOLO 模型以每秒 45 帧的速度实时处理图像。一个更小版本的网络 Fast YOLO,每秒处理惊人的 155 帧,同时仍然实现了其他实时检测器两倍的 mAP。与最先进的检测系统相比,YOLO 在定位错误上更多,但在预测不存在物体的误检方面可能性小得多。最后,YOLO 学习了非常通用的物体表示。在从自然图像泛化到毕加索数据集和人物艺术数据集的艺术作品时,它大幅超越了所有其他检测方法,包括 DPM 和 R-CNN。

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 far less likely to predict false detections where nothing exists. Finally, YOLO learns very general representations of objects. It outperforms all other detection methods, including DPM and R-CNN, by a wide margin when generalizing from natural images to artwork on both the Picasso Dataset and the People-Art Dataset.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 16)

阅读逐段中英对照全文 →