Facebook AI发布新的计算机视觉库Detectron2

2020-04-01 13:23:24    来源:    作者:

Facebook AI Research(FAIR)发布了Detectron2,这是一个基于PyTorch的计算机视觉库,为流行的框架带来了一系列新的研究和生产功能。

Facebook AI发布新的计算机视觉库Detectron2

自2018年发布以来,原始的Detectron对象检测平台已成为FAIR最广泛采用的开源项目之一。虽然第一个Detectron是用Caffe2编写的,但Detectron2代表了PyTorch中从头开始对原始框架的完全重写,并具有一些新的对象检测功能。

在首次发布时,Detron成为AI社区的巨大推动力。它使许多人可以快速,轻松地构建最新的对象检测模型。然而Detectron仍然存在一些局限性,这些局限性很快成为许多AI从业者的难题。

Detectron2旨在解决这些交易突破问题,从而打造了一个更强大,更现代的图书馆。来自FAIR的Detectron2团队:

Facebook AI发布新的计算机视觉库Detectron2

我们构建了Detectron2,以满足Facebook AI的研究需求,并为Facebook生产用例中的对象检测提供了基础。现在,我们正在使用Detectron2快速设计和训练下一代姿态检测模型,这些模型可为Smart Camera(Facebook门户视频通话设备中的AI摄像机系统)提供动力。通过将Detectron2用作用于研究和生产用例中对象检测的统一库,我们能够将研究思路迅速转移到大规模部署的生产模型中。

转向PyTorch符合AI社区对灵活但易于使用的库的不断增长的需求和渴望。PyTorch本身在设计上是模块化的,因此比Caffe2易于扩展。AI社区中的绝大多数已经使用了两个库:TensorFlow和PyTorch。

Detectron2已扩展到处理计算机视觉任务,而不仅仅是对象检测,包括语义分割,全景分割,姿势估计和DensePose。作者已经做出了令人瞩目的努力,以添加经过训练的最新模型,例如Cascade R-CNN,Panoptic FPN和TensorMask。

Facebook AI发布新的计算机视觉库Detectron2

FAIR的团队在其官方博客文章中暗示,他们计划向库中发布一个附加组件Detectron2go,以使其更易于将模型部署到生产中。据说它包括网络量化,模型优化和移动部署格式设置等功能。

郑重声明:本文版权归原作者所有,转载文章仅为传播更多信息之目的,如作者信息标记有误,请第一时间联系我们修改或删除,多谢。