mask-rcnn 环境搭建

一、简介

  windows平台的mask-rcnn环境搭建(keras版)

二、下载地址

  • 项目下载:
    源码:https://github.com/matterport/Mask_RCNN
  • coco预训练权重下载:
    https://github.com/matterport/Mask_RCNN/releases/download/v2.0/mask_rcnn_coco.h5

三、依赖库

  keras==2.2.4
  tensorflow-gpu==1.10.0
  numpy==1.14.5
  scipy==1.3.1
  Pillow==6.1.0
  cython==0.29.13
  matplotlib==3.1.1
  scikit-image==0.15.0
  opencv-python==4.1.1.26
  h5py==2.10.0
  imgaug0.3.0
  IPython
7.8.0

四、常见问题

1、pip安装imgaug时报错

mask-rcnn 环境搭建

(1)下载shapely

  通过这个网址https://www.lfd.uci.edu/~gohlke/pythonlibs/#shapely找到适合自己python版本的shapely下载到本地,我使用的64位操作系统、python3.6 ,下载Shapely-1.6.4.post2-cp36-cp36m-win_amd64.whlmask-rcnn 环境搭建

(2)安装shapely

  进入whl文件所在的文件夹,执行以下命令
  pip install Shapely-1.6.4.post2-cp36-cp36m-win_amd64.whl

(3)再次尝试安装imgaug

  pip intall imgaug

2、pip安装coco报错

mask-rcnn 环境搭建
  目前官方COCO 并不支持在 Windows 上安装。有开发者对 COCO 做了一些修改,可以支持windows。我们从github上下载支持windows的源码来进行安装。有两种安装方式可供选择:

(1)使用pip命令进行安装

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

(2)下载源码进行安装

  --在 https://github.com/philferriere/cocoapi 下载源码,并进行解压。
  --以管理员身份打开 CMD 终端,并切换到 *\cocoapi-master\PythonAPI 目录。
  --运行以下指令:
  python setup.py build_ext install
  --运行以上指令时如果出现以下错误提示:
mask-rcnn 环境搭建
  --解决方法:此种安装方法需要使用 Microsoft Visual C++ 14.0 对 COCO 源码进行编译。如果本地不支持 Microsoft Visual C++ 14.0 或者版本低于 14.0,可以通过安装 Microsoft Visual Studio 2015 及以上版本。

(3)查看cocoapi安装是否成功

  安装成功,可以看在python的三方lib中(路径通常在lib\site-packages)看到pycocotools
mask-rcnn 环境搭建

3、使用import win32api时,报错:No module named win32api

(1)使用pip命令进行安装

  pip install pypiwin32或pip3 install pypiwin32 或 python -m pip install pypiwin32

(2)下载源码进行安装

  --在 https://www.lfd.uci.edu/~gohlke/pythonlibs/#pywin32 找到适合自己python版本的pywin32下载下来
  --进入whl文件所在的文件夹,执行以下命令:
  pip install pywin32‑225‑cp36‑cp36m‑win_amd64.whl