首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Facebook 发布 Detectron2:基于 PyTorch 的新一代目标检测工具

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

作者头像
AI研习社
发布2019-10-16 20:42:26
1K0
发布2019-10-16 20:42:26
举报
文章被收录于专栏:AI研习社AI研习社

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

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

Detectron2 简介

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

Detectron2 与之前的 Detectron 不同,它并非在原版基础上进行改写,而是彻底重写的全新版本。新版目标检测平台 Detectron2 则基于 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: 新增了将模型产品化部署的软件实现,包括:标准的内部数据训练工作流实现、模型压缩量化、模型转化等。
本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2019-10-14,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 AI研习社 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
相关产品与服务
图像识别
腾讯云图像识别基于深度学习等人工智能技术,提供车辆,物体及场景等检测和识别服务, 已上线产品子功能包含车辆识别,商品识别,宠物识别,文件封识别等,更多功能接口敬请期待。
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档