目前最全:52 个深度学习目标检测模型汇总,论文、源码一应俱全!

221 阅读5分钟

文章来源:zhuanlan.zhihu.com/p/115035951

作者:红色石头\

目标检测作为计算机视觉中的一个重要分支,近些年来随着神经网络理论研究的深入和硬件 GPU 算力的大幅度提升,一举成为全球人工智能研究的热点,落地项目也最先开始。

纵观 2013 年到 2020 年,从最早的 R-CNN、OverFeat 到后来的 SSD、YOLO v3 再到去年的 M2Det,新模型层出不穷,性能也越来越好!本文将完整地总结 52 个目标检测模型极其性能对比,包括完备的文献 paper 列表。

首先直奔主题,列出这 52 个目标检测模型(建议收藏):

这份表看得太舒适了!这份目标检测超全的技术路线总结来自于 GitHub 上一个知名项目,作者是毕业于韩国首尔国立大学电气与计算机工程专业的 Lee hoseong,目前已经收获 7.6k star。

该项目地址是:

github.com/hoya012/dee…

该技术路线纵贯的时间线是 2013 年到 2020 年初,上图总结了这期间目标检测所有具有代表性的模型。图中标红的部分是相对来说比较重要,需要重点掌握的模型。

更新日志

值得一提的是红色石头早在去年年初的时候已经发文给大家推荐过这个项目,作者也一直在更新,截至 2020 年 2 月,作者主要的更新如下:

  • 2019.2:更新3篇论文
  • 2019.3:更新图表和代码链接
  • 2019.4:更新 ICLR 2019 和 CVPR 2019 论文
  • 2019.5:更新 CVPR 2019 论文
  • 2019.6:更新 CVPR 2019 论文和数据集论文
  • 2019.7:更新 BMVC 2019 论文和部分 ICCV 2019 论文
  • 2019.9:更新 NeurIPS 2019 论文和 ICCV 2019 论文
  • 2019.11:更新部分 AAAI 2020 论文和其它论文
  • 2020.1:更新 ICLR 2020 论文和其它论文

下面详细介绍!

模型性能对比表

由于硬件不同(例如 CPU、GPU、RAM 等),来比较 FPS 往往不够准确。更合适的比较方法是在同一硬件配置下测量所有模型的性能。以上所有模型的性能对比结果如下:

从上面的表格中,可以清楚看到不同模型在 VOC07、VOC12、COCO 数据集上的性能表现。同时列出了模型论文发表来源。\

下面列举一些重点标红的模型进行简要介绍。

模型论文篇

2014 年\

R-CNN

Rich feature hierarchies for accurate object detection and semantic segmentation | [CVPR' 14]

论文:

arxiv.org/pdf/1311.25…

官方代码 Caffe:

github.com/rbgirshick/…

OverFeat

OverFeat: Integrated Recognition, Localization and Detection using Convolutional Networks | [ICLR' 14]

论文:

arxiv.org/pdf/1312.62…

官方代码 Torch:

github.com/sermanet/Ov…

2015 年

Fast R-CNN

Fast R-CNN | [ICCV' 15]

论文:

arxiv.org/pdf/1504.08…

官方代码 caffe:

github.com/rbgirshick/…

Faster R-CNN

Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks | [NIPS' 15]

论文:

papers.nips.cc/paper/5638-…

官方代码 caffe:

github.com/rbgirshick/…

非官方代码 tensorflow:

github.com/endernewton…

非官方代码 pytorch:

github.com/jwyang/fast…

2016 年

OHEM

Training Region-based Object Detectors with Online Hard Example Mining | [CVPR' 16]

论文:

arxiv.org/pdf/1604.03…

官方代码 caffe:

github.com/abhi2610/oh…

YOLO v1

You Only Look Once: Unified, Real-Time Object Detection | [CVPR' 16]

论文:

arxiv.org/pdf/1506.02…

官方代码 c:

pjreddie.com/darknet/yol…

SSD

SSD: Single Shot MultiBox Detector | [ECCV' 16]

论文:

arxiv.org/pdf/1512.02…

官方代码 caffe:

github.com/weiliu89/ca…

非官方代码 tensorflow:

github.com/balancap/SS…

非官方代码 pytorch:

github.com/amdegroot/s…

R-FCN

R-FCN: Object Detection via Region-based Fully Convolutional Networks | [NIPS' 16]

论文:

arxiv.org/pdf/1605.06…

官方代码 caffe:

github.com/daijifeng00…

非官方代码 caffe:

github.com/YuwenXiong/…

2017 年\

YOLO v2

YOLO9000: Better, Faster, Stronger | [CVPR' 17]

论文:

arxiv.org/pdf/1612.08…

官方代码 c:

pjreddie.com/darknet/yol…

非官方代码 caffe:

github.com/quhezheng/c…

非官方代码 tensorflow:

github.com/nilboy/tens…

非官方代码 tensorflow:

github.com/sualab/obje…

非官方代码 pytorch:

github.com/longcw/yolo…

FPN

Feature Pyramid Networks for Object Detection | [CVPR' 17]

论文:

openaccess.thecvf.com/content_cvp…

非官方代码 caffe:

github.com/unsky/FPN

RetinaNet

Focal Loss for Dense Object Detection | [ICCV' 17]

论文:

arxiv.org/pdf/1708.02…

官方代码 keras:

github.com/fizyr/keras…

非官方代码 pytorch:

github.com/kuangliu/py…

非官方代码 mxnet:

github.com/unsky/Retin…

非官方代码 tensorflow:

github.com/tensorflow/…

Mask R-CNN

Mask R-CNN | [ICCV' 17]

论文:

openaccess.thecvf.com/content_ICC…

官方代码 caffe2:

github.com/facebookres…

非官方代码 tensorflow:

github.com/matterport/…

非官方代码 tensorflow:

github.com/CharlesShan…

非官方代码 pytorch:

github.com/multimodall…

2018 年

YOLO v3

YOLOv3: An Incremental Improvement | [arXiv' 18]

论文:

pjreddie.com/media/files…

官方代码 c:

pjreddie.com/darknet/yol…

非官方代码 pytorch:

github.com/ayooshkathu…

非官方代码 pytorch:

github.com/eriklindern…

非官方代码 keras:

github.com/qqwweee/ker…

非官方代码 tensorflow:

github.com/mystic123/t…

RefineDet

Single-Shot Refinement Neural Network for Object Detection | [CVPR' 18]

论文:

openaccess.thecvf.com/content_cvp…

官方代码 caffe:

github.com/sfzhang15/R…

非官方代码 chainer:

github.com/fukatani/Re…

非官方代码 pytorch:

github.com/lzx1413/Pyt…

2019 年

M2Det

M2Det: A Single-Shot Object Detector based on Multi-Level Feature Pyramid Network | [AAAI' 19]

论文:

arxiv.org/pdf/1811.04…

官方代码 pytorch:

github.com/qijiezhao/M…

2020 年

Spiking-YOLO

Spiking-YOLO: Spiking Neural Network for Real-time Object Detection | [AAAI' 20]

论文:

arxiv.org/pdf/1903.06…

数据集论文篇

同时作者也列出了以上模型通常使用的公开数据集:VOC、ILSVRC、COCO,如下表所示:

用于目标检测的数据集相关论文如下:

以上就是 52 个深度学习检测模型汇总介绍。该项目可以说把近几年的目标检测模型总结得很不错了,包括论文和源码。希望对大家有所帮助!

往期精彩回顾




适合初学者入门人工智能的路线及资料下载机器学习在线手册深度学习在线手册AI基础下载(pdf更新到25集)本站qq群1003271085,加入微信群请回复“加群”获取一折本站知识星球优惠券,请回复“知识星球”喜欢文章,点个在看