1、Mask_RCNN源代码下载,地址如下:
https://github.com/matterport/Mask_RCNN
安装步骤见源代码官方给出的办法。
我单独安装的tensorflow-gpu和keras故在requirements.txt文件中删去了这两部分,再执行如下:
pip3 install -r requirements.txt --user
再运行安装命令python3 setup.py install --user
注:这里我安装的是Python3.6,参考博客
并安装相应依赖库:numpy,scipy,cython,h5py,Pillow,scikit-image, opencv-python,imgaug,keras等。通过pip3 install numpy --user
格式的命令安装。
安装keras之前需要安装tensorflow-gpu版本,如2、3步骤的安装过程;
2、安装GPU版本的tensorflow,通过如下指令安装:
pip3 install --upgrade https://mirrors.tuna.tsinghua.edu.cn/tensorflow/linux/gpu/tensorflow_gpu-1.4.0rc1-cp36-cp36m-linux_x86_64.whl --user
3、安装keras,这里安装的是Keras2.1,pip3 install Keras==2.1 --user
4、下载预先训练好的COCO模型权重 mask_rcnn_coco.h5,下载链接为
https://github.com/matterport/Mask_RCNN/releases
下载好的权重文件,放在Mask_RCNN文件根目录下。
5、安装pycocotools库
git clone https://github.com/waleedka/coco.git
pip3 install git+https://github.com/waleedka/coco.git#subdirectory=PythonAPI --user
或者采用如下方法,也可以安装:
在安装时吧对应文件夹下的python改为python3,便可安装在python3的库下,
$ git clone https://github.com/waleedka/coco.git
$ cd coco-master/PythonAPI
$ sudo make install
接下来,吧makefile里面的python改为了python3(因为我是用的是python3.6)。
6、将demo.ipynb转成了demo.py,放在和demo.ipynb相同的地方,这里的转换需要下载安装Jupyter进行转换,可以直接在vscode中安装插件来进行转换。抓换好之后直接运行python文件会报错如下:
File "demo.py", line 35, in <module>
get_ipython().magic(u'matplotlib inline')
NameError: name 'get_ipython' is not defined
这是因为get_ipython()函数只能在Jupyter环境下使用,在python环境中无法使用,解决办法很直接,注释该行#get_ipython().magic(u'matplotlib inline')
再次执行python3 demo.py
出现如下运行效果:
至此,代码算是运行成功了!
参考博客:
https://blog.csdn.net/qq_38589460/article/details/85332207
https://blog.csdn.net/qujiaqi1030/article/details/99675432