我々は、物体検出のための新しいアプローチであるYOLOを提示する。これまでの物体検出の研究では、検出を実行するために分類器を再利用していました。その代わりに、我々は物体検出を空間的に分離されたバウンディングボックスと関連するクラス確率への回帰問題としてフレーム化します。単一のニューラルネットワークは、1回の評価でフル画像から直接バウンディングボックスとクラス確率を予測します。検出パイプライン全体が単一のネットワークであるため、検出性能を直接エンドツーエンドで最適化することができます。当社の統一されたアーキテクチャは非常に高速です。ベースとなるYOLOモデルは、毎秒45フレームでリアルタイムに画像を処理します。ネットワークの小型バージョンであるFast YOLOは、他のリアルタイム検出器の2倍のmAPを達成しながら、155フレーム/秒という驚異的な処理を行います。最先端の検出システムと比較して、YOLOはより多くの定位エラーを発生させますが、何も存在しない場所での誤検出を予測する可能性ははるかに低くなっています。最後に、YOLOは物体の非常に一般的な表現を学習します。Picasso DatasetとPeople-Art Datasetの両方で、自然画像からアートワークに一般化する場合、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.