（1）mask-rcnn之Demo运行（win10&ubuntu）

 

目录

 

 

1.依赖

2.下载源码(python3.4+)

3.下载训练模型文件，并放在主目录下面

4.运行demo

4.1 启动jupyter-notebook

4.2 打开simple/demo.ipynb，修改一下代码

5.运行demo

6.可能会出现的问题

---

1.依赖

安装tensorflow和keras由于版本不兼容走了不少弯路。目前可使用一下版本：

 pip install tensorflow==1.9.0 --user

 pip install keras==2.0.8 --user

安装pycocotools

 pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI

2.下载源码(python3.4+)

# 克隆源码
git clone https://github.com/matterport/Mask_RCNN.git
# 进入项目目录
cd Mask_RCNN
# 安装依赖
pip3 install -r requirements.txt
# 安装项目
python setup.py install

3.下载训练模型文件，并放在主目录下面

1. 在https://github.com/matterport/Mask_RCNN/releases下载mask_rcnn_coco.h5
2. 也可以在百度网盘下载：链接: https://pan.baidu.com/s/1XQfwuCrHUR9PwK28jXSuGg 提取码: dhf7 

注意：如果没有科学上网，在github下载文件可能不完整，会报“OSError: Unable to open file (truncated file: eof = 24614474, sblock->base_addr = 0, stored_eof = 257557808)：”的错误，我在这里卡了很久，惭愧了！

4.运行demo

4.1 启动jupyter-notebook

# 进入项目主目录
cd Mask_RCNN

# 打开jupyter-notebook
jupyter-notebook

4.2 打开simple/demo.ipynb，修改一下代码

模型文件名称

修改这两个变量，不然会报“len(images) must be equal to BATCH_SIZE”，这里看了一下源码：self.BATCH_SIZE = self.IMAGES_PER_GPU * self.GPU_COUNT

5.运行demo

点击cell--运行全部,即可看到结果

6.可能会出现的问题

1. 缺各种包，缺什么安装就好。
2. tensorflow，keras中提示没有moudle，按照步骤1安装兼容版本的tensorflow，keras