EfficientDet目标检测谷歌官方终于开源了!

因为结果吸引人,Github上已经有多份非官方开源实现,这次谷歌大脑的官方版本终于来了。

开源地址:

https://github.com/google/automl/tree/master/efficientdet

位于google新开的automl项目内,看样子以后这个项目还会有其他自动机器学习的算法开源。

EfficientDet原出于论文 EfficientDet: Scalable and Efficient Object Detection,开源页面显示,这篇论文已经被CVPR 2020接收。

在算法设计上有三大特点:

1. 优秀的主干网络。同样出自谷歌家族的EfficientNets 。

2. 双向FPN(BiFPN,特征金字塔网络)。可以方便且更好的进行特征融合。

3. 模型缩放技术。设计好模型的主干网络、特征网络、预测网络后,按照一定的优化规则,在网络的深度、宽度、输入图像的分辨率上进行模型缩放,故EfficientDet其实是一系列网络,可在统一架构下得到适合移动端和追求高精度的多个模型。

作者们结合BiFPN和特征融合策略设计了与YOLOv3精度相仿的EfficientDet-D0,使用模型缩放技术得到一系列检测模型:EfficientDet-D1 到 D6,在精度和模型复杂度上权衡。

其高精度模型 EfficientDet-D6 在COCO数据集上达到 50.9 mAP,而仅需要51.9M 参数 和 229B FLOPs。相比于之前的最好算法(AmoebaNet + NAS-FPN + AutoAugment )达到了更高的精度,却仅有1/4参数量,1/13的FLOPs,在GPU/CPU上运行结果快3~5倍!

这次开源诚意满满,提供了EfficientDet D0-D6所有预训练模型:

同时为了方便使用,谷歌也提供了训练代码。

(0)

相关推荐

  • PFLD:高精度实时人脸关键点检测算法

    重磅干货,第一时间送达 本文转自:AI算法与图像处理 这篇文章作者分别来自天津大学.武汉大学.腾讯AI实验室.美国天普大学.该算法对在高通ARM 845处理器可达140fps:另外模型大小较小,仅2. ...

  • 2021年小目标检测最新研究综述

    作者丨高新波.莫梦竟成.汪海涛.冷佳旭  编辑丨极市平台 极市导读 与以往将小目标与常规目标等同对待或只关注特定应用场景下的目标检测综述不同,本文对小目标检测这一不可或缺且极具挑战性的研究领域进行了系 ...

  • 目标检测算法21篇速览:检测网络优化及改进

    作者丨SDC 来源丨GiantPandaCV 编辑丨极市平台 极市导读 本文总结了21篇目标检测算法方面的论文,包括对已有的两种检测网络设计范式的调整和优化,在检测网络中添加注意力模块的方式和方法以及 ...

  • 综述:目标检测二十年(2001-2021)

    作者丨派派星 来源丨CVHub 编辑丨极市平台 极市导读 目标检测领域发展至今已有二十余载,本文将对目标检测领域近20年的发展做一次系统性的介绍,并为目标检测构建一个完整的知识体系架构. 引言 目标检 ...

  • 基于机器视觉技术的奶牛跛行自动识别研究进展

    内容节选自: 韩书庆, 张晶, 程国栋, 彭英琦, 张建华, 吴建寨. 奶牛跛行自动识别技术研究现状与挑战[J]. 智慧农业(中英文), 2020, 2(3): 21-36. HAN Shuqing, ...

  • 基于手机系统的实时目标检测

    计算机视觉研究院专栏 作者:Edison_G 在具有有限计算能力和存储器资源的移动设备上运行卷积神经网络(CNN)模型的日益增长的需求促进了对有效模型设计的研究 公众号ID|ComputerVisio ...

  • 基于Adaboost算法的并网光伏发电系统的孤岛检测法

    ★中国电工技术学会出品★ 致力于产业界与学术界融合创新的品牌会议 ①浏览会议通知,请戳下面标题 ☟ ☞第二轮通知︱2018第十二届中国电工装备创新与发展论坛暨第八届电工技术前沿问题学术论坛 ②参会注册 ...

  • 多媒体内容理解在美图社区的应用实践

    编辑整理:翁梦娟 出品平台:DataFunTalk 导读:移动互联网时代,图像和短视频等多媒体内容爆发,基于计算机视觉的AI算法是多媒体内容分析的基础.在美图社区智能化发展的过程中,视频和图像分类打标 ...

  • 谷歌40人发表59页长文:为何真实场景中ML模型表现不好?

    大家好哇,前不久,在卖萌屋NLP群里默默潜水的白鹡鸰被群友提到的一篇Google几天前放出的59页超长论文炸得飞了起来. 来,大家来感受一下气势浩大的论文首页 文章名字是Underspecificat ...

  • 用预训练卷积网络提取图像重要 so easy !

    "   横看成岭侧成峰,远近高低各不同. 不识庐山真面目,只缘身在此山中." 开头一首诗,很装啊!老铁 我们都知道各种卷积网络提取特征什么鬼的,好像很吊的样子. 那到底提取处理的会 ...

  • 【学术论文】一种改进的RefineDet多尺度人脸检测方法

    摘要: 针对车站.商场等大型场所中客流量大.背景复杂等原因导致多尺度人脸检测精度低的问题,建立了一种基于RefineDet多层特征图融合的多尺度人脸检测方法.首先利用第一级网络进行特征提取并在不同尺度 ...

  • 基于Mask-GD分割的机器人抓取检测

    重磅干货,第一时间送达 小黑导读 论文是学术研究的精华和未来发展的明灯.小黑决心每天为大家带来经典或者最新论文的解读和分享,旨在帮助各位读者快速了解论文内容.个人能力有限,理解难免出现偏差,建议对文章 ...

  • 什么是目标检测中的平均精度均值(mAP)?

    重磅干货,第一时间送达 计算机视觉界已经集中在度量 mAP 上,来比较目标检测系统的性能.在这篇文章中,我们将深入了解平均精度均值 (mAP) 是如何计算的,以及为什么 mAP 已成为目标检测的首选指 ...

  • YOLO v4:物体检测的最佳速度和精度

    重磅干货,第一时间送达 YOLOYOLO v4 1 介绍 将YOLOv3的AP和FPS分别提高10%和12%[5](浅蓝色区域的模型被视为实时目标检测器) 可以看出,EfficientDet D4-D ...