网易首页 > 网易号 > 正文 申请入驻

可以在手机里运行的Detectron2来了:FB官方出品

0
分享至

  

鱼羊 发自 凹非寺
量子位 报道 | 公众号 QbitAI

  做目标检测、语义分割,你一定听说过Detectron2。

  作为一个基于PyTorch实现的模块化目标检测库,Detectron2当年刚一开源,就被推上了GitHub趋势榜第一。

  而现在,移动端开发人员们的福利来了。

  Facebook官方正式推出Detectron2的移动版:Detectron2Go(D2Go)。

  

  什么是D2Go

  先来看官方介绍:

  

  • PyTorch和Detectron2支持的深度学习工具包
  • 最先进的高效移动设备骨干网络
  • 支持端到端模型训练、量化和部署
  • 能轻松导出TorchScript格式

  通过D2Go,开发者可以创建已经针对移动设备进行过优化的FBNet模型,在移动端高效地执行目标检测语义分割以及关键点估计等任务。

  

  △D2Go人体关键点估计

  那么D2Go具体有哪些优势呢?

  实际上,移动端的目标检测任务主要关注的有两点:延迟准确性

  而如果模型能够在边缘设备上独立运行,不用将数据传至云端进行处理,就能大大的减少延迟。

  另外,这也进一步保障了终端用户的数据、隐私安全。

  Facebook表示,在实验测试中,使用D2Go开发的移动端模型,与基于服务器的模型相比,延迟更低,且准确性损失不大。

  如何使用

  首先是安装的部分,需要安装的有:

  PyTorch Nightly

  conda install pytorch torchvision cudatoolkit=10.2 -c pytorch-nightly

  Detectron2

  python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'

  mobile_cv

  python -m pip install 'git+https://github.com/facebookresearch/mobile-vision.git'

  D2go

  git clone https://github.com/facebookresearch/d2go
cd d2go & python -m pip install .

  在D2go的GitHub仓库中,还提供了预训练模型的推理Demo。

  从model_zoo中选择一个模型,运行demo.py,就可以进行试玩。

  以faster_rcnn_fbnetv3a模型为例:

  cd demo/
python demo.py —config-file faster_rcnn_fabnetv3a_C4.yaml —input input1.jpg —output output1.jpg

  在训练和评估方面,D2Go本身基于detectron2工具包实现,因此在训练之前,需要按照detectron2的说明设置内置数据集。

  而具体如何用预训练模型进行推理训练一个D2go模型将模型导出到int8,Facebook也提供了详细的入门示例。

  

  开源地址:
https://github.com/facebookresearch/d2go

  Facebook博客:
https://ai.facebook.com/blog/d2go-brings-detectron2-to-mobile/

特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相关推荐
热点推荐

俄军大兵压境后,泽林斯基将怨气撒到港企身上,下令冻结一切资产

科技精灵
2021-04-11 21:26:18

这3种蔬菜,炒前必须“焯水”,专家:再懒也不能省略这一步!

阳仔饮食shi
2021-04-11 20:22:45

现在起,这些罪名不归公安机关管!这下全说清了

生活直通车
2021-04-10 00:58:02

延迟退休方案出炉,70后看完后松了口气,最受影响的是这批人

西凉铁马
2021-04-10 18:28:11

啥情况?“雪藏”20年后“重见天日”

记录之路
2021-04-11 02:32:37

涉嫌严重违纪违法!南通一干部接受审查调查

启东热线
2021-04-11 20:41:16

北控输辽宁!两将停赛,杨被死草,替补+孙悦+罚球全崩,张帆受伤

篮球资讯达人
2021-04-11 21:52:56

胡适的“父母无恩论”,真的有道理吗?

有料观史
2021-04-11 20:40:02

惊奇队长换淑女妆容,金发波浪露靓肩,网友却在争论造型女权问题

猫眼电影
2021-04-11 16:11:31

预计从7月中下旬开始,中心城区将有30到42天中到重度伏旱

五彩斑斓的世界
2021-04-11 21:22:07

缅甸影帝别迪乌和影后恩达觉欣夫妇被逮捕

缅甸中文网
2021-04-10 19:50:40

2021最新手机销量排名:iPhone12全球第一,前十没有华为

北境不忘
2021-04-10 17:56:28

公安局员工被羁押489天三次被判刑,无罪后获国赔20余万

澎湃新闻
2021-04-11 19:44:01

央视发布:喜欢跑步需要知道的9件事

跑吧
2021-04-11 09:00:02

厅官落马情妇为90后 涉案财物曝光:25件奢侈品中10个爱马仕包

态度山东
2021-04-10 16:39:19

1-0!2-1!逆转+绝杀!西班牙人克星踢疯了,武磊被保送冲甲+争冠

超级替补
2021-04-11 09:06:26

来自地狱的美食藤壶,到底有多可怕?看鲸鱼被折磨的样子就懂了

你要的我都有
2021-04-10 12:41:54

卖净水器的总说自来水不干净,而自来水公司为何也一直默默无声?

一姐说历史
2021-04-10 16:37:11

董明珠问岳云鹏:你赚的钱没郭麒麟赚的多吧?岳云鹏回答让人笑翻

娱最资讯
2021-04-10 09:47:01

保安儿子考上名校要借5万,老板拒绝立即报警,接过信封愣住了!

素锦流光
2021-04-11 10:18:29
2021-04-12 00:05:07
量子位
量子位
追踪人工智能动态
4751文章数 95979关注度
往期回顾 全部

科技要闻

小米决定造车始末:特朗普推了雷军一把?

头条要闻

“元气森林”欺骗消费者 饮料中其实并非无糖

头条要闻

“元气森林”欺骗消费者 饮料中其实并非无糖

体育要闻

15投11铁!郭艾伦复出手感凉 多亏了队友

娱乐要闻

刘诗诗穿薄荷绿裙秀香肩 温柔清新

财经要闻

汽车要闻

工况续航达419km 奔腾NAT推5款网约车

态度原创

时尚
艺术
亲子
本地
数码

专访王菊:人要性感,穿要大胆

艺术要闻

中国美术学院“望道讲谈”

亲子要闻

5岁孩子看完就去马桶舀水喝!《小猪佩奇》等21部动画片被查出问题

本地新闻

致敬理想燃烧与命运突围的日子

数码要闻

华为专业办公显示器最新爆料:采用高刷屏