Facebook 发布 Detectron2:基于 PyTorch 的新一代目标检测工具

来源: 赢家财富网 作者:佚名

摘要: Detectron 是 FAIR 在 2018 年初公开的目标检测平台,包含了大量业内最具代表性的目标检测、图像分割、关键点检测算法,该框架主要基于 python 和 caffe2 实现,开源项目已获得了超 2.2w 的星标数。

  Detectron 是 FAIR 在 2018 年初公开的目标检测平台,包含了大量业内最具代表性的目标检测、图像分割、关键点检测算法,该框架主要基于 python 和 caffe2 实现,开源项目已获得了超 2.2w 的星标数。

  而 Detectron2 则是刚推出的新版 PyTorch 1.3  中一重大新工具,它源于 maskrcnn 基准测试,但它却是对先前版本 detectron 的一次彻底重写。Detectron2 包含了更大的灵活性与扩展性,并增强了可维护性和可伸缩性,以支持在生产中的用例。目前已在 GitHub 上开源,雷锋网 AI 开发者将其重要内容整理编译如下。

Detectron2 简介

  据官方介绍,Detectron2 是 FAIR 的下一代目标检测和分割研究平台。目前,它实现了最先进的目标检测算法。

  Detectron 2 与之前的 detectron 不同,它并非在原版基础上进行改写,而是彻底重写的全新版本。新版目标检测平台 Detectron 2 则基于 PyTorch 框架,以 maskrcnn 基准测试作为起点。通过全新的模块化设计,Detectron2 变得更灵活且易于扩展,它能够在单个或多个 GPU 服务器上提供更快速的训练速度。

Detectron2 的新特性

  基于 PyTorch 框架 与 Detectron 在 Caffe2 中实现不同,Detectron2 则基于 PyTorch 实现。PyTorch 提供了一个更直观的命令式编程模型,它允许研究人员和工程师可以更快地迭代模型设计和实验,并让开发者受益于 PyTorch 深度学习方法以及不断改进的 PyTorch 活跃社区。

  模块化、可扩展设计 在 Detectron2 中,引入了模块化设计,用户可以将自定义模块插入目标检测系统的任意部分。这意味着许多新的研究项目可以用数百行代码编写,而核心 Detectron2 库和全新的研究之间可以实现完全分离。通过建立新模型并发现新方法,可以不断完善模块化和可扩展的设计,这也帮助 Detectron2 变得更加灵活。其模块化特性也能够使其更有效地帮助研究人员探索最先进的算法设计。

  新模型和新功能 Detectron2 包括原始 Detectron 中的所有模型,例如 Faster R-CNN,Mask R-CNN,RetinaNet 和 DensePose。除此之外,Detectron2 还加入了几个新模型,包括:Cascade R-CNN,Panoptic FPN 和 TensorMask,未来还将继续添加更多算法。在 Detectron2 还包含了一些新功能,例如:同步批处理规范化(synchronous Batch Norm),以及对 LVIS 等新数据集的支持等。

  新任务支持 Detectron2 支持与目标检测有关的一系列任务。和之前的 Detectron 一样,它也支持目标检测、实例分割、人体姿态估计等任务。此外,Detectron2 还支持语义分割和全景分割。

  高实现质量 重写的 Detectron2 能够重新检测底层的设计决策,并解决原来 Detectron 的几个实现问题。

  速度和可扩展性 通过将整个训练 pipeline 移至 GPU,能够使 Detectron2 比原来的 Detectron 更快,而且可以更加方便在各种标准型号下进行 GPU 服务器的分布式训练,从而轻松扩展训练数据集。

  Detectron2go 新增了将模型产品化部署的软件实现,包括:标准的内部数据训练工作流实现、模型压缩量化、模型转化等。


审核:yj149 编辑:yj149
关键词:

Facebook,检测工具,目标,Detectron2

免责声明:

1:凡本网注明“来源:***”的作品,均是转载自其他平台,本网赢家财富网 www.yjcf360.com 转载文章为个人学习、研究或者欣赏传播信息之目的,并不意味着赞同其观点或其内容的真实性已得到证实。全部作品仅代表作者本人的观点,不代表本网站赢家财富网的观点、看法及立场,文责作者自负。如因作品内容、版权和其他问题请与本站管理员联系,请在30日内进行,我们收到通知后会在3个工作日内及时进行处理。

2:本网站刊载的各类文章、广告、访问者在本网站发表的观点,以链接形式推荐的其他网站内容,仅为提供更多信息供用户参考使用或为学习交流的方便(本网有权删除)。所提供的数据仅供参考,使用者务请核实,风险自负。

版权属于赢家财富网,转载请注明出处
查看更多