mask_rcnn

环境的搭建

安装 Mask_Rcnn

下载源码

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

安装依赖

安装前先修改 requirements.txt 这个文件。

diff --git a/requirements.txt b/requirements.txt
index bd64785..eabf100 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,12 +1,12 @@
-numpy
+numpy==1.16.1
 scipy
 Pillow
 cython
 matplotlib
 scikit-image
-tensorflow>=1.3.0
-keras>=2.0.8
+tensorflow==1.5.0
+keras==2.1.4
 opencv-python

pip3 install -r requirements.txt

安装 Mask_Rcnn

python3 setup.py install

根据readme 下载训练好的数据集 mask_rcnn_coco.h5

安装demo依赖库

git clone https://github.com/cocodataset/cocoapi.git

cd cocoapi/PythonAPI

make

make install

运行 demo

将 Mask_RCNN/sample/demo.ipynb 下的 .ipynb 文件转换成 . py 文件

pip3 install jupyter

jupyter nbconvert --to script xxx.ipynb 

++这里遇到一个奇怪的问题。就是install之后需要重启电脑才能使用 jupyter 这个指令。且仅在终端里面有用，SSH里面还是会报错。找不到这个命令++

python demo.py

如果报错 get_ipython().run_line_magic(‘matplotlib’,‘inline’)
把这行注释掉即可