查看: 1665|回复: 0

配置 Mask-RCNN (matterport)

[复制链接]

665

主题

1234

帖子

6568

积分

xdtech

Rank: 5Rank: 5

积分
6568
发表于 2019-2-9 00:21:28 | 显示全部楼层 |阅读模式
此文记录配置 Mask-RCNN 的相关工作,具体为 matterport/Mask_RCNN ,系统环境为 Ubuntu 16.04 。
准备工作matterport/Mask_RCNN 中系统及依赖要求为:
  • Python 3.4+
  • TensorFlow 1.3+
  • Keras 2.0.8+
  • Jupyter Notebook
  • Numpy, skimage, scipy, Pillow, cython, h5py
使用 Anaconda 新建环境,用以配置 Mask-RCNN:
# 创建一个名为 Mask-RCNN 的环境,制定 Python 版本为3.6conda create --name Mask-RCNN python=3.6然后在新建的环境内分别安装其他依赖包即可。 TensorFlow 建议参照官方安装流程,当然,多半会发现因为 Wall 的原因导致安装失败。 可以使用清华源作为替代: TensorFlow 镜像使用帮助  
直接使用 conda 命令安装的 TensorFlow 基本无法成功调用 GPU,没有 GPU 则可以考虑,毕竟方便。
配置 Mask-RCNN下载/Clone matterport/Mask_RCNN或者在  releases page 中下载。
下载 mask_rcnn_coco.h5
Download pre-trained COCO weights (mask_rcnn_coco.h5) from the releases page.
安装 pycocotools
  • 下载源文件后打开 coco/PythonAPI ,并在此目录下打开终端,进入之前新建的 Mask-RCNN 环境,运行 make 。
  • 如果终端提示 Mask-RCNN 环境中缺少 Cython ,则重新安装后运行 make。
  • 将生成的 pycocotools 文件夹复制到 Mask-RCNN 的源文件中即可。



    Screenshotfrom2018-03-0718-54-11.png

  • 使用 Jupyter 运行 demo.ipynb 即可看到预测的图样:





  • 如果存在无法调用 coco 的情况,可以考虑将 pycocotools 路径(coco/PythonAPI)添加到系统路径。



作者:DexterLei
链接:https://www.jianshu.com/p/f51339f886fc
來源:简书
简书著作权归作者所有,任何形式的转载都请联系作者获得授权并注明出处。

回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表