Tensorflow + Keras 运行Mask-RCNN

源码地址：https://github.com/matterport/Mask_RCNN

实验环境：Ubuntu + tensorflow1.4 + cuda8.0 + python3.5 + Keras2.1

 

前几天在tensorflow上测试了MaskRCNN，见https://blog.csdn.net/BockSong/article/details/82690343

但这个repo已经较长时间不维护了，效果也不是很理想。于是选用了github上star数最多的MaskRCNN项目，在keras和tensorflow上再次进行实验。

相关的博客资料已经较为丰富，配置过程中主要参考了：

https://blog.csdn.net/chenmoran0928/article/details/79999073

https://blog.csdn.net/wlnvgu/article/details/78489346

这里对配置中遇到的问题做一下记录。

1、关于notebook

在还没配完环境的时候提前打开了notebook，装完后后测试时报错，后来发现是notebook环境没有更新。需要在命令行里重启jupyter notebook，再次从头运行demo文件，即可顺利载入模块。

2、关于pycocotools的安装

首先下载项目源码 https://github.com/waleedka/coco （所在路径无要求）

根据另一篇博客所述，如果是Ubuntu系统，PythonAPI编译时需先用文本编辑器打开makefile，把python改成python3

然后执行如下指令（权限不够的话加上sudo）

make
make install
python setup.py install

3、生成model时提示：TypeError: softmax() got an unexpected keyword argument 'axis'

查找后发现是Keras新版和repo中的旧版语法冲突了，新版Keras去掉了这一参数。因此回退Keras版本到2.1。新版Keras附带了preprocessing和applications两个工具，为了卸载干净我先单独卸掉这两个工具，再执行如下命令重装旧版本。

pip install keras==2.1

之后demo即可顺利运行。