Mask-RCNN环境配置及代码调试

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值