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

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

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


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


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


image.png

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


该项目地址是:

https://github.com/hoya012/deep_learning_object_detection


该技术路线纵贯的时间线是 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 往往不够准确。更合适的比较方法是在同一硬件配置下测量所有模型的性能。以上所有模型的性能对比结果如下:


11.jpg

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


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


模型论文篇


2014 年


R-CNN


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


论文:

https://arxiv.org/pdf/1311.2524.pdf


官方代码 Caffe:

https://github.com/rbgirshick/rcnn


OverFeat

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


论文:

https://arxiv.org/pdf/1312.6229.pdf


官方代码 Torch:

https://github.com/sermanet/OverFeat


2015 年


Fast R-CNN


Fast R-CNN | [ICCV' 15]


论文:

https://arxiv.org/pdf/1504.08083.pdf


官方代码 caffe:

https://github.com/rbgirshick/fast-rcnn


Faster R-CNN


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


论文:

https://papers.nips.cc/paper/5638-faster-r-cnn-towards-real-time-object-detection-with-region-proposal-networks.pdf


官方代码 caffe:

https://github.com/rbgirshick/py-faster-rcnn


非官方代码 tensorflow:

https://github.com/endernewton/tf-faster-rcnn


非官方代码 pytorch:

https://github.com/jwyang/faster-rcnn.pytorch


2016 年


OHEM


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


论文:

https://arxiv.org/pdf/1604.03540.pdf


官方代码 caffe:

https://github.com/abhi2610/ohem


YOLO v1


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


论文:

https://arxiv.org/pdf/1506.02640.pdf


官方代码 c:

https://pjreddie.com/darknet/yolo/


SSD


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


论文:

https://arxiv.org/pdf/1512.02325.pdf


官方代码 caffe:

https://github.com/weiliu89/caffe/tree/ssd


非官方代码 tensorflow:

https://github.com/balancap/SSD-Tensorflow


非官方代码 pytorch:

https://github.com/amdegroot/ssd.pytorch


R-FCN


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


论文:

https://arxiv.org/pdf/1605.06409.pdf


官方代码 caffe:

https://github.com/daijifeng001/R-FCN


非官方代码 caffe:

https://github.com/YuwenXiong/py-R-FCN


2017 年


YOLO v2


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


论文:

https://arxiv.org/pdf/1612.08242.pdf


官方代码 c:

https://pjreddie.com/darknet/yolo/


非官方代码 caffe:

https://github.com/quhezheng/caffe_yolo_v2


官方代码 tensorflow:

https://github.com/nilboy/tensorflow-yolo


官方代码 tensorflow:

https://github.com/sualab/object-detection-yolov2


官方代码 pytorch:

https://github.com/longcw/yolo2-pytorch


FPN


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


论文:

http://openaccess.thecvf.com/content_cvpr_2017/papers/Lin_Feature_Pyramid_Networks_CVPR_2017_paper.pdf


非官方代码 caffe:

https://github.com/unsky/FPN


RetinaNet


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


论文:

https://arxiv.org/pdf/1708.02002.pdf


官方代码 keras:

https://github.com/fizyr/keras-retinanet


非官方代码 pytorch:

https://github.com/kuangliu/pytorch-retinanet


非官方代码 mxnet:

https://github.com/unsky/RetinaNet


非官方代码 tensorflow:

https://github.com/tensorflow/tpu/tree/master/models/official/retinanet


Mask R-CNN


Mask R-CNN | [ICCV' 17]


论文:

http://openaccess.thecvf.com/content_ICCV_2017/papers/He_Mask_R-CNN_ICCV_2017_paper.pdf


官方代码 caffe2:

https://github.com/facebookresearch/Detectron


非官方代码 tensorflow:

https://github.com/matterport/Mask_RCNN


非官方代码 tensorflow:

https://github.com/CharlesShang/FastMaskRCNN


非官方代码 pytorch:

https://github.com/multimodallearning/pytorch-mask-rcnn


2018 年


YOLO v3


YOLOv3: An Incremental Improvement | [arXiv' 18]


论文:

https://pjreddie.com/media/files/papers/YOLOv3.pdf


官方代码 c:

https://pjreddie.com/darknet/yolo/


非官方代码 pytorch

https://github.com/ayooshkathuria/pytorch-yolo-v3


非官方代码 pytorch:

https://github.com/eriklindernoren/PyTorch-YOLOv3


非官方代码 keras:

https://github.com/qqwweee/keras-yolo3


非官方代码 tensorflow:

https://github.com/mystic123/tensorflow-yolo-v3


RefineDet


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


论文:

http://openaccess.thecvf.com/content_cvpr_2018/papers/Zhang_Single-Shot_Refinement_Neural_CVPR_2018_paper.pdf


官方代码 caffe:

https://github.com/sfzhang15/RefineDet


非官方代码 chainer:

https://github.com/fukatani/RefineDet_chainer


非官方代码 pytorch:

https://github.com/lzx1413/PytorchSSD


2019 年


M2Det


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


论文:

https://arxiv.org/pdf/1811.04533.pdf


官方代码 pytorch:

https://github.com/qijiezhao/M2Det


2020 年


Spiking-YOLO


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


论文:

https://arxiv.org/pdf/1903.06530.pdf


数据集论文篇


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


111.jpg

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


image.png

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

相关文章
|
2月前
|
机器学习/深度学习 算法 定位技术
Baumer工业相机堡盟工业相机如何通过YoloV8深度学习模型实现裂缝的检测识别(C#代码UI界面版)
本项目基于YOLOv8模型与C#界面,结合Baumer工业相机,实现裂缝的高效检测识别。支持图像、视频及摄像头输入,具备高精度与实时性,适用于桥梁、路面、隧道等多种工业场景。
240 0
|
3月前
|
机器学习/深度学习 监控 算法
基于mediapipe深度学习的手势数字识别系统python源码
本内容涵盖手势识别算法的相关资料,包括:1. 算法运行效果预览(无水印完整程序);2. 软件版本与配置环境说明,提供Python运行环境安装步骤;3. 部分核心代码,完整版含中文注释及操作视频;4. 算法理论概述,详解Mediapipe框架在手势识别中的应用。Mediapipe采用模块化设计,包含Calculator Graph、Packet和Subgraph等核心组件,支持实时处理任务,广泛应用于虚拟现实、智能监控等领域。
|
2月前
|
机器学习/深度学习 人工智能 PyTorch
AI 基础知识从 0.2 到 0.3——构建你的第一个深度学习模型
本文以 MNIST 手写数字识别为切入点,介绍了深度学习的基本原理与实现流程,帮助读者建立起对神经网络建模过程的系统性理解。
212 15
AI 基础知识从 0.2 到 0.3——构建你的第一个深度学习模型
|
2月前
|
机器学习/深度学习 人工智能 自然语言处理
AI 基础知识从 0.3 到 0.4——如何选对深度学习模型?
本系列文章从机器学习基础出发,逐步深入至深度学习与Transformer模型,探讨AI关键技术原理及应用。内容涵盖模型架构解析、典型模型对比、预训练与微调策略,并结合Hugging Face平台进行实战演示,适合初学者与开发者系统学习AI核心知识。
231 15
|
2月前
|
机器学习/深度学习 存储 监控
基于深度学习YOLO框架的城市道路损伤检测与评估项目系统【附完整源码+数据集】
本项目基于深度学习的YOLO框架,成功实现了城市道路损伤的自动检测与评估。通过YOLOv8模型,我们能够高效地识别和分类路面裂缝、井盖移位、坑洼路面等常见的道路损伤类型。系统的核心优势在于其高效性和实时性,能够实时监控城市道路,自动标注损伤类型,并生成损伤评估报告。
139 0
基于深度学习YOLO框架的城市道路损伤检测与评估项目系统【附完整源码+数据集】
|
2月前
|
机器学习/深度学习 自动驾驶 算法
基于深度学习的YOLO框架的7种交通场景识别项目系统【附完整源码+数据集】
在智慧交通和智能驾驶日益普及的今天,准确识别复杂交通场景中的关键元素已成为自动驾驶系统的核心能力之一。传统的图像处理技术难以适应高动态、复杂天气、多目标密集的交通环境,而基于深度学习的目标检测算法,尤其是YOLO(You Only Look Once)系列,因其检测速度快、精度高、可部署性强等特点,在交通场景识别中占据了重要地位。
281 0
基于深度学习的YOLO框架的7种交通场景识别项目系统【附完整源码+数据集】
|
2月前
|
机器学习/深度学习 人工智能 自然语言处理
深度学习模型、算法与应用的全方位解析
深度学习,作为人工智能(AI)的一个重要分支,已经在多个领域产生了革命性的影响。从图像识别到自然语言处理,从语音识别到自动驾驶,深度学习无处不在。本篇博客将深入探讨深度学习的模型、算法及其在各个领域的应用。
350 4
|
3月前
|
机器学习/深度学习 存储 PyTorch
PyTorch + MLFlow 实战:从零构建可追踪的深度学习模型训练系统
本文通过使用 Kaggle 数据集训练情感分析模型的实例,详细演示了如何将 PyTorch 与 MLFlow 进行深度集成,实现完整的实验跟踪、模型记录和结果可复现性管理。文章将系统性地介绍训练代码的核心组件,展示指标和工件的记录方法,并提供 MLFlow UI 的详细界面截图。
103 2
PyTorch + MLFlow 实战:从零构建可追踪的深度学习模型训练系统
|
7月前
|
机器学习/深度学习 数据采集 自然语言处理
深度学习实践技巧:提升模型性能的详尽指南
深度学习模型在图像分类、自然语言处理、时间序列分析等多个领域都表现出了卓越的性能,但在实际应用中,为了使模型达到最佳效果,常规的标准流程往往不足。本文提供了多种深度学习实践技巧,包括数据预处理、模型设计优化、训练策略和评价与调参等方面的详细操作和代码示例,希望能够为应用实战提供有效的指导和支持。
|
8月前
|
机器学习/深度学习 运维 安全
深度学习在安全事件检测中的应用:守护数字世界的利器
深度学习在安全事件检测中的应用:守护数字世界的利器
310 22

热门文章

最新文章