如何用Facebook的免费平台Detectron进行物体检测?

姓名:闫伟  学号:15020150038

转载自:https://zhuanlan.zhihu.com/p/27847470,有删节。

【嵌牛导读】:Detectron是Facebook FAIR开源了的一个目标检测(Object Detection)平台。

【嵌牛鼻子】:Facebook Detectron

【嵌牛提问】:那么用 Facebook 的 Detectron 平台该怎么检测物体呢?

【嵌牛正文】:

今年 1 月份,Facebook 对外开源了旗下物体检测平台 Detectron,其基于一个深度学习框架,目前支持大量的机器学习算法。它的开源意味着 Facebook 人工智能实验室在物体检测领域使用的软件,像我们这些普通开发者也能用了,也为我们提供了 TensorFlow 之外的另一款强大工具。Detectron 的其中一个先进功能就是物体掩蔽(object masking),它比物体检测更进一步,不再是围绕图像画出一个简单的边界框,而是画出一个复杂的多边形,比如下图检测“卡戴珊姐妹”的效果:

下面就为大家展示如何使用 Detectron 执行物体检测任务。

安装 Caffe2

第一步就是先安装Caffe2

git clone --recursive https://github.com/caffe2/caffe2.git && cd caffe2make && cd build && sudo make installpython -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"

注:要有英伟达 GPU 和 CUDA,才能使用 Detectron。

安装 COCO API

git clone https://github.com/cocodataset/cocoapi.gitcd cocoapi/PythonAPImake install

克隆仓库

成功安装 Caffe2 后,需要克隆Detectron 的仓库

git clone https://github.com/facebookresearch/Detectron.git && cd Detectron

安装所需的依赖:

pip install \numpy \pyyaml \matplotlib \opencv-python>=3.0 \setuptools \Cython \mock \Scipy

创建 Python 模块:

cd lib && make && cd ..

创建推断(inference)

现在我们只需使用如下命令,对仓库中的 demo 照片进行物体检测:

python tools/infer_simple.py \--cfg configs/12_2017_baselines/e2e_mask_rcnn_R-101-FPN_2x.yaml \--output-dir demo/output \--image-ext jpg \--wts \https://s3-us-west-2.amazonaws.com/detectron/35861858/12_2017_baselines/e2e_mask_rcnn_R-101-FPN_2x.yaml.02_32_51.SgT4y1cO/output/train/coco_2014_train:coco_2014_valminusminival/generalized_rcnn/model_final.pkl \demo

然后你就能在 Detectron/demo/output 目录下发现一张 PDF 格式的照片输出。

如果想测试你自己的照片,只需往 Detectron/demo 目录中添加 JPG 文件。最好让照片像素在 600-800px 之间,这样效果最佳。

总的来看,Detectron 对一般物体不管是进行检测还是掩蔽,都有不错的表现。如果你想玩转除了 TensorFlow 之外的其它物体检测平台,不妨试试 Detectron。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 219,589评论 6 508
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 93,615评论 3 396
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 165,933评论 0 356
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,976评论 1 295
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,999评论 6 393
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,775评论 1 307
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,474评论 3 420
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,359评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,854评论 1 317
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 38,007评论 3 338
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 40,146评论 1 351
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,826评论 5 346
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,484评论 3 331
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 32,029评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,153评论 1 272
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,420评论 3 373
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 45,107评论 2 356