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

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


目标检测(Object Detection)是深度学习 CV 领域的一个核心研究领域和重要分支。纵观 2013 年到 2019 年,从最早的 R-CNN、Fast R-CNN 到后来的 YOLO v2、YOLO v3 再到今年的 M2Det,新模型层出不穷,性能也越来越好!本文将会对目标检测近几年的发展和相关论文做出一份系统介绍,总结一份超全的文献 paper 列表。


模型列表先一睹为快!(建议收藏


image.png


这份目标检测超全的技术路线总结来自于 GitHub 上一个知名项目,作者是 Lee hoseong,项目地址是:


https://github.com/hoya012/deep_learning_object_detection


该技术路线横跨时间是 2014 年至 2019 年,上图总结了这期间目标检测所有重要的模型。图中标红的部分是作者认为比较重要,需要重点掌握的模型。当然每个人有都有各自的评价。


模型性能比较


FPS(速度)索引与硬件规格(如 CPU、GPU、RAM 等)有关,因此很难进行同等比较。解决方案是在具有相同规格的硬件上测量所有模型的性能,但这是非常困难和耗时的。比较结果如下:


image.png

image.png


下面举例对标红的重要模型进行介绍!


2014 年


R-CNN


Rich feature hierarchies for accurate object detection and semantic segmentation | Ross Girshick, Jeff Donahue, Trevor Darrell, Jitendra Malik | [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 | Pierre Sermanet, et al. | [ICLR' 14]


论文:

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


代码 Torch:

https://github.com/sermanet/OverFeat


2015 年


Fast R-CNN


Fast R-CNN | Ross Girshick | [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 | Shaoqing Ren, et al. | [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 | Abhinav Shrivastava, et al. | [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 | Joseph Redmon, et al. | [CVPR' 16]


论文:

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


代码 c:

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


SSD


Single Shot MultiBox Detector | Wei Liu, et al. | [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


Object Detection via Region-based Fully Convolutional Networks | Jifeng Dai, et al. | [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


Better, Faster, Stronger | Joseph Redmon, Ali Farhadi | [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 | Tsung-Yi Lin, et al. | [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 | Tsung-Yi Lin, et al. | [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


Kaiming He, et al. | [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


An Incremental Improvement | Joseph Redmon, Ali Farhadi | [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 | Shifeng Zhang, et al. | [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


A Single-Shot Object Detector based on Multi-Level Feature Pyramid Network | Qijie Zhao, et al. | [AAAI' 19]


论文:

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


参考文献


该项目的参考文献来自于论文《Deep Learning for Generic Object Detection: A Survey


论文地址:


https://arxiv.org/pdf/1809.02165v1.pdf

相关文章
|
5天前
|
机器学习/深度学习 数据采集 TensorFlow
使用Python实现智能食品市场预测的深度学习模型
使用Python实现智能食品市场预测的深度学习模型
30 5
|
4天前
|
机器学习/深度学习 人工智能 自然语言处理
探索深度学习中的Transformer模型
探索深度学习中的Transformer模型
12 1
|
7天前
|
机器学习/深度学习 算法 开发者
探索深度学习中的优化器选择对模型性能的影响
在深度学习领域,优化器的选择对于模型训练的效果具有决定性作用。本文通过对比分析不同优化器的工作原理及其在实际应用中的表现,探讨了如何根据具体任务选择合适的优化器以提高模型性能。文章首先概述了几种常见的优化算法,包括梯度下降法、随机梯度下降法(SGD)、动量法、AdaGrad、RMSProp和Adam等;然后,通过实验验证了这些优化器在不同数据集上训练神经网络时的效率与准确性差异;最后,提出了一些基于经验的规则帮助开发者更好地做出选择。
|
6天前
|
机器学习/深度学习 算法 数据可视化
使用Python实现深度学习模型:智能食品配送优化
使用Python实现深度学习模型:智能食品配送优化
22 2
|
5天前
|
机器学习/深度学习 人工智能 算法
【手写数字识别】Python+深度学习+机器学习+人工智能+TensorFlow+算法模型
手写数字识别系统,使用Python作为主要开发语言,基于深度学习TensorFlow框架,搭建卷积神经网络算法。并通过对数据集进行训练,最后得到一个识别精度较高的模型。并基于Flask框架,开发网页端操作平台,实现用户上传一张图片识别其名称。
21 0
【手写数字识别】Python+深度学习+机器学习+人工智能+TensorFlow+算法模型
|
5天前
|
机器学习/深度学习 人工智能 算法
基于深度学习的【蔬菜识别】系统实现~Python+人工智能+TensorFlow+算法模型
蔬菜识别系统,本系统使用Python作为主要编程语言,通过收集了8种常见的蔬菜图像数据集('土豆', '大白菜', '大葱', '莲藕', '菠菜', '西红柿', '韭菜', '黄瓜'),然后基于TensorFlow搭建卷积神经网络算法模型,通过多轮迭代训练最后得到一个识别精度较高的模型文件。在使用Django开发web网页端操作界面,实现用户上传一张蔬菜图片识别其名称。
25 0
基于深度学习的【蔬菜识别】系统实现~Python+人工智能+TensorFlow+算法模型
|
8天前
|
机器学习/深度学习 数据采集 TensorFlow
使用Python实现智能食品储存管理的深度学习模型
使用Python实现智能食品储存管理的深度学习模型
23 2
|
8天前
|
机器学习/深度学习 人工智能 测试技术
深度学习在图像识别中的应用与挑战
本文探讨了深度学习技术,尤其是卷积神经网络(CNN)在图像识别任务中的最新进展和面临的主要挑战。通过分析不同的网络架构、训练技巧以及优化策略,文章旨在提供一个全面的概览,帮助研究人员和实践者更好地理解和应用这些技术。
37 9
|
4天前
|
机器学习/深度学习 人工智能 算法
深度学习在图像识别中的应用与挑战
本文探讨了深度学习技术在图像识别领域的应用,重点分析了卷积神经网络(CNN)的工作原理及其在处理图像数据方面的优势。通过案例研究,展示了深度学习如何提高图像识别的准确性和效率。同时,文章也讨论了当前面临的主要挑战,包括数据不足、过拟合问题以及计算资源的需求,并提出了相应的解决策略。
|
5天前
|
机器学习/深度学习 分布式计算 并行计算
深度学习在图像识别中的应用与挑战
本文深入探讨了深度学习技术在图像识别领域的应用,分析了当前主流的卷积神经网络(CNN)架构,并讨论了在实际应用中遇到的挑战和可能的解决方案。通过对比研究,揭示了不同网络结构对识别准确率的影响,并提出了优化策略。此外,文章还探讨了深度学习模型在处理大规模数据集时的性能瓶颈,以及如何通过硬件加速和算法改进来提升效率。