从零开始 Mask RCNN 实战:基于 Win10 + Anaconda 的 Mask RCNN 环境搭建

简介: 从零开始 Mask RCNN 实战:基于 Win10 + Anaconda 的 Mask RCNN 环境搭建

大名鼎鼎的 Mask RCNN 一举夺得 ICCV2017 Best Paper,名声大造。Mask RCNN 是何恺明基于以往的 Faster RCNN 架构提出的新的卷积网络,实现高效地检测图中的物体,并同时生成一张高质量的每个个体的分割掩码,即有效地目标的同时完成了高质量的语义分割。


image.png


论文地址:

https://arxiv.org/abs/1703.06870


开源地址:

https://github.com/matterport/Mask_RCNN


本文主要介绍如何配置 Mask RCNN 的训练和测试环境,为了方便起见,选择的外部环境是 Win10 + Anaconda。


1. 安装 Anaconda


安装 Anaconda 很简单,在其官网上下载对应的 Win10(64位)版本,一步一步安装即可。安装后在启动栏显示安装成功的 Anaconda 组件。


image.png


2. 下载 Mask RCNN 开源库


从 GitHub 网站:

https://github.com/matterport/Mask_RCNN


将该项目下载下来。


同时下载 Mask RCNN 的预训练模型 “mask_rcnn_coco.h5”,放置于本地 Mask_RCNN 开源库的根目录下。


“mask_rcnn_coco.h5” 下载地址:

https://github.com/matterport/Mask_RCNN/releases


在里面的 Mask R-CNN 2.0 下找到 “mask_rcnn_coco.h5” 并下载。


3. 创建虚拟环境


在 Mask RCNN 源码目录下的 README.md,查看环境要求:

打开 Anaconda Prompt:


image.png


如图, Python 版本要求在 3.4 以上,TensorFlow 版本要求在 1.3 以上,Keras 版本要求在 2.0.8 以上。

image.png

image.png

这样直接安装,由于网络问题可能会出现安装失败的情况。解决方法是使用国内的镜像源下载,例如清华、豆瓣。


清华:

https://pypi.tuna.tsinghua.edu.cn/simple


豆瓣:

https://pypi.douban.com/simple


CPU 安装命令改为:

image.png

image.png

安装完成之后,可以验证。进入 Python 环境,输入 “import tensorflow as tf” 和 “import keras”,如果没有报错,则证明 TensorFlow 和 Keras 均安装成功。


image.png


6. 安装 requirements.txt 中的所有 Python 库


在刚下载的 Mask RCNN 源代码的根目录下有一个 requirements.txt,里面包含了所需的所有 Python 库。因此,我们还要安装:


  • numpy
  • scipy
  • Pillow
  • cython
  • matplotlib
  • scikit-image
  • opencv-python
  • h5py
  • imgaug
  • IPython[all]


安装方法也很简单,例如安装 numpy:



pip install -i https://pypi.douban.com/simple numpy


其他 Python 库的安装类似。


7. 安装 pycocotools 库


如果需要在 COCO 数据集上训练或测试,需要安装 pycocotools(coco 数据集的应用 API)。如果你不需要在 COCO 数据集上训练和测试,只使用 Mask RCNN 训练自己的数据集,则可以直接跳过这一步。


1)下载 pycocotools 源文件


github下载:

https://github.com/philferriere/cocoapi


码云下载:

https://gitee.com/ACANX/cocoapi


2)提前准备


安装 Cython(之前在 requirements.txt 中已经安装过了)


安装 VS2015 环境


因为之前我已经安装整个 VS 2015 了,所以这个环境不缺,可参考:


https://blog.csdn.net/sinat_33486980/article/details/92840377


3)安装


  • 打开 Anaconda Prompt,进入 MaskRCNN 环境;
  • 切换到 cocoapi\PythonAPI 目录;
  • 依次运行:



python setup.py build_ext --inplace


python setup.py build_ext install

进入 Python shell,输入以下命令,验证 pycocotools 是否安装成功:

image.png


8. 验证环境搭建是否成功


下面使用 Mask RCNN 源代码提供的 demo 程序,验证一下环境是否搭建成功。


首先,打开 Anaconda Prompt,进入 MaskRCNN 环境,安装 jupyter notebook,方法与上面安装其它 Python 库类似,也可以在 Anaconda Navigator 中直接安装。


然后,在 Anaconda Prompt 中,输入 “jupyter notebook”,打开 jupyter notebook。


接着,打开 Mask RCNN 源代码 samples 目录下的 demo.ipynb 文件。


image.png

最后,选择 Cell 菜单,在 Cell 下拉菜单选择 Run All,稍等片刻,在该页面底部会输出运行结果:


image.png


大功告成!基于 Win10 + Anaconda 的 MaskRCNN 环境搭建顺利完成~


相关文章
|
人工智能 计算机视觉 Python
【YOLOv8-Seg】实战二:LabVIEW+OpenVINO加速YOLOv8-seg实例分割
【YOLOv8-Seg】实战二:LabVIEW+OpenVINO加速YOLOv8-seg实例分割
542 0
【YOLOv8-Seg】实战二:LabVIEW+OpenVINO加速YOLOv8-seg实例分割
|
机器学习/深度学习 计算机视觉 异构计算
Darknet53详细原理(含torch版源码)
Darknet53详细原理(含torch版源码)—— cifar10
559 0
Darknet53详细原理(含torch版源码)
|
机器学习/深度学习 存储 JSON
YOLOv5的Tricks | 【Trick10】从PyTorch Hub加载YOLOv5
YOLOv5的Tricks | 【Trick10】从PyTorch Hub加载YOLOv5
1232 0
YOLOv5的Tricks | 【Trick10】从PyTorch Hub加载YOLOv5
|
计算机视觉 索引
YOLOv5的Tricks | 【Trick14】YOLOv5的val.py脚本的解析
YOLOv5的Tricks | 【Trick14】YOLOv5的val.py脚本的解析
1463 0
YOLOv5的Tricks | 【Trick14】YOLOv5的val.py脚本的解析
|
机器学习/深度学习 人工智能 计算机视觉
【YOLOv8-seg】实战一:手把手教你使用YOLOv8实现实例分割
【YOLOv8-seg】实战一:手把手教你使用YOLOv8实现实例分割
4616 0
【YOLOv8-seg】实战一:手把手教你使用YOLOv8实现实例分割
|
机器学习/深度学习 人工智能 计算机视觉
【YOLOv8】实战二:YOLOv8 OpenVINO2022版 windows部署实战
【YOLOv8】实战二:YOLOv8 OpenVINO2022版 windows部署实战
1276 0
【YOLOv8】实战二:YOLOv8 OpenVINO2022版 windows部署实战
|
机器学习/深度学习 算法 数据挖掘
【检测|RCNN系列3】原理+源码彻底搞懂Faster RCNN(文末下载论文)
【检测|RCNN系列3】原理+源码彻底搞懂Faster RCNN(文末下载论文)
1667 0
|
移动开发 并行计算 IDE
【MASK-RCNN配置】Windows环境下配置完整方案
一、配置python环境 1下载Python Python Releases for Windows | Python.org 2下载PyCharm Download PyCharm: Python IDE for Professional Developers by JetBrains 2.OpenCV_Python 软件环境的搭建:为人脸检测,人脸采集,人眼检测项目做准备_哔哩哔哩_bilibili ............
609 0
|
机器学习/深度学习 算法 PyTorch
手把手教你使用LabVIEW实现Mask R-CNN图像实例分割(含源码)
前面给大家介绍了使用LabVIEW工具包实现图像分类,目标检测,今天我们来看一下如何使用LabVIEW实现Mask R-CNN图像实例分割。
277 0
|
异构计算
【超快超轻YOLO】YOLO-Fastest从Darknet源码编译、测试再到训练完整图文教程!
【超快超轻YOLO】YOLO-Fastest从Darknet源码编译、测试再到训练完整图文教程!
505 0