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
IPython7.8.0
四、常见问题
1、pip安装imgaug时报错
(1)下载shapely
通过这个网址https://www.lfd.uci.edu/~gohlke/pythonlibs/#shapely找到适合自己python版本的shapely下载到本地,我使用的64位操作系统、python3.6 ,下载Shapely-1.6.4.post2-cp36-cp36m-win_amd64.whl
(2)安装shapely
进入whl文件所在的文件夹,执行以下命令
pip install Shapely-1.6.4.post2-cp36-cp36m-win_amd64.whl
(3)再次尝试安装imgaug
pip intall imgaug
2、pip安装coco报错
目前官方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
--运行以上指令时如果出现以下错误提示:
--解决方法:此种安装方法需要使用 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
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