中文

YOLO9000:更好、更快、更强

计算机视觉与模式识别 2016-12-28 v1

摘要

我们引入了 YOLO9000,这是一个 SOTA 的实时目标检测系统,能够检测超过 9000 个目标类别。首先,我们提出了对 YOLO 检测方法的各种改进,既有新颖的改进,也有借鉴自先前工作的成果。改进后的模型 YOLOv2 在 PASCAL VOC 和 COCO 等标准检测任务上达到了 SOTA。在 67 FPS 下,YOLOv2 在 VOC 2007 上取得了 76.8 mAP。在 40 FPS 下,YOLOv2 取得了 78.6 mAP,在运行速度显著更快的同时,超越了 Faster RCNN with ResNet 和 SSD 等 SOTA 方法。最后,我们提出了一种联合训练目标检测与分类的方法。使用该方法,我们在 COCO 检测数据集和 ImageNet 分类数据集上同时训练了 YOLO9000。我们的联合训练使 YOLO9000 能够对没有标注检测数据的目标类别进行检测预测。我们在 ImageNet 检测任务上验证了该方法。尽管在 200 个类别中仅有 44 个类别的检测数据,YOLO9000 在 ImageNet 检测验证集上仍取得了 19.7 mAP。在不在 COCO 中的 156 个类别上,YOLO9000 取得了 16.0 mAP。但 YOLO 能检测的远不止 200 个类别;它能预测超过 9000 种不同目标类别的检测结果,并且依然保持实时运行。

关键词

引用

@article{arxiv.1612.08242,
  title  = {YOLO9000: Better, Faster, Stronger},
  author = {Joseph Redmon and Ali Farhadi},
  journal= {arXiv preprint arXiv:1612.08242},
  year   = {2016}
}