首先需要下载 mask_rcnn 的源码
但是从GitHub上下载源码的速度非常慢,所以我们从码云上下载,这是GitHub的中国镜像。
链接如下:https://gitee.com/mirrors/Mask_RCNN?_from=gitee_search
下载pycocotools
什么需要安装pycocotools,经过看源码发现,训练coco数据集时用到了pycocotools这个模块,如果不安装会报错无法正常运行。
到 https://github.com/philferriere/cocoapi 下载源码,并进行解压。以管理员身份打开 CMD 终端,并切换到 cocoapiPythonAPI目录。运行以下指令:
#install pycocotools locally
python setup.py build_ext --inplace
#install pycocotools to the Python site-packages
python setup.py build_ext install
下载预先训练的COCO权重(mask_rcnn_coco.h5)
- 这个可以从GitHub下载
- 或者留下邮箱,我给你发过去
- 或者去CSDN的下载资源去下载
- 下载后放在 mask_rcnn 的根目录
安装
- CD 进入 mask_rcnn 项目根目录
pip install -r requirements.txt
- 仍是在根目录运行安装程序
python3 setup.py install
(不报错的话就安装完成了,如果报错可以根据错误提示,网络搜索解决。python3 不行的话就用 python。
还要注意一点你使用哪个python环境安装,后面运行的时候也要用此python环境运行MaskRCNN)
演示
- cd 到 Mask_RCNNsamples目录
- 用安装Mask RCNN的python环境打开 jupyter notebook,命令行,或shell运行:
jupyter notebook
- 打开 samples/demo.ipynb 文件,运行所有cell,这样就可以运行demo了。
- 可以更换images目录里面的图片,用自己的图片进行测试