Mask_RCNN 使用记录

Mask_RCNN  demo 使用:

1/ 下载:git clone https://github.com/matterport/Mask_RCNN

2/  下载权重文件:https://github.com/matterport/Mask_RCNN/releases

     选择 Mask R-CNN 2.0 中 mask_rcnn_coco.h5 文件下载

3/ 使用pycharm 打开 Mask_RCNN-master/samples/demo.ipynb 文件,因为自己用不惯(不懂)notebook ,所以将demo.ipynb中的代码copy到了 新建的demo.py 文件中:

 因为路径不一样要更改以下几个地方,将以下2行代码注释:

# import coco

# %matplotlib inline (这行不知道是什么作用直接注释)

增加:

from samples.coco import coco

在COCO_MODEL_PATH 和 IMAGE_DIR 中加入  'Mask_RCNN-master' :因为我的路径是这样,大家根据自己路径更改

IMAGE_DIR = os.path.join(ROOT_DIR,'Mask_RCNN-master', "images")
COCO_MODEL_PATH = os.path.join(ROOT_DIR,'Mask_RCNN-master', "mask_rcnn_coco.h5")

将自己想要测试的图片放入 images 文件夹 然后就可以运行了:

 

 

Mask_RCNN 使用记录