ubuntu下MASK-RCNN运行记录

ubuntu下MASK-RCNN运行记录

一，环境

1.ubuntu18.04 (tensorflow>=1.3.0)
2.tensorflow (tensorflow>=1.3.0)
3.keras>=2.0.8

二，下载代码

git clone https://github.com/matterport/Mask_RCNN

注意下载时要下载完全。

三，安装Python coco API

git clone https://github.com/pdollar/coco.git
cd coco/PythonAPI
make

make后会生成文件夹如图

将编译（make）后的pycocotools整个目录拷贝到mask程序根目录下

四，安装依赖项

先cd到mask的根目录下，有一个requirements.txt，再输入：

pip3 install -r requirements.txt

五，修改demo

源文件demo后缀为.ipynb，需要修改为.py后缀名。
1.打开Ipython Notebook

python -m IPython notebook

2.打开需要转换的 .ipynb 文件
需要修改的demo文件在Mask_RCNN/samples文件夹中
3.修改后保存并复制到Mask_RCNN/samples文件夹中

六，下载权重文件

百度网盘下载：
链接：https://pan.baidu.com/s/1drKvfg 密码：yer9
手动下载：https://github.com/matterport/Mask_RCNN/releases/download/v2.0/mask_rcnn_coco.h5

下载后放到根目录下

七，修改demo.py

1.修改权重路径

路径修改为自己权重的保存位置。
2.注释掉：

get_ipython().run_line_magic('matplotlib', 'inline')

3.运行demo.py文件