Mask-RCNN简单实现
1.下载Mask-RCNN框架
学习深度学习的话还是比较推荐用Ubuntu系统的,毕竟大家都是工程应用。这篇教程是在Ubuntu系统下运行的。
这篇文章默认已经装好tensorflow和Cuda。
git clone https://github.com/matterport/Mask_RCNN
下载好后需要一个已经训练好的模型,放在这个文件夹下。
这里给出百度网盘下载地址
链接:https://pan.baidu.com/s/1drKvfg 密码:yer9
2.安装依赖库文件
Mask-RCNN要求的PY依赖库
numpy
scipy
Pillow
cython
matplotlib
scikit-image
tensorflow>=1.3.0
keras>=2.0.8
opencv-python
h5py
imgaug
IPython[all]
一个个安装,然后官方的程序是在jupyter-notebook这个框架下跑的,也可以把代码转成.py格式在pycharm下跑也没有什么问题。
或者直接
pip3 install -r requirements.txt
3.安装测试
上面步骤没有什么大问题就可以直接测试运行一下下
python setup.py
基本上没报错问题就不大了,如果报错也不用慌,因为我第一次安装也是报错,不过我运行demo也没什么问题。
4.运行官方的sample
前提是已经把训练好的模型放好在Mask-RCNN的文件夹下面,没有的话则会报错,看第一个步骤。
jupyter-notebook
运行jupyter,打开Mask-RCNN目录下的sample文件夹,直接点开demo.ipynb文件。
然后先一步步运行,如果也库文件没安装就先退出pip安装一下再进去。
5.检测自己的图片
在Mask-RCNN的文件夹下面,打开image这个文件夹,里面就是放置待检测的图片,如果你想检测自己的图片,可以先把里面的图片移除到另外的地方,再把自己的图片放进去。
对自己的图片没有大小尺寸要求,可以直接把图片放进去,就可以看到检测的结果。
下一篇就会讲怎么训练自己的模型然后测试