Mask_RCNN官方示例算法链接
https://github.com/matterport/Mask_RCNN#getting-started
环境:tf_py365 (python3.6.5)
jupyter notebook中运行demo.ipynb
1 安装skimage.io
(安装前运行会报错:ModuleNotFoundError: No module named ‘skimage’)
pip install scikit-image
2 安装imageaug
(安装前运行会报错:ModuleNotFoundError: No module named ‘imgaug’)
pip install imageaug
3 安装cython(python加速工具)
(安装前运行会报错error: Unable to find vcvarsall.bat等)
conda install cython
4 安装VS编译好的pycocotools
(安装前运行报错:ModuleNotFoundError: No module named ‘pycocotools’)
网上搜索VS编译好的pycocotools,下载,放入指定目录下:
(1)python工程目录
(2)anaconda/虚拟环境/lib/site-packages
两种方法都可以from pycocotools.coco import COCO。路径(1)因为是放到工程里,所以这个工程在所有的python3.6、3.5环境下都可以使用pycocotools;路径(2)是放到虚拟环境里,其他工程加载环境就可以直接用。
(python 3.7环境或无法使用)
若不安装编译好的pycocotools,可官网下载源码,在VS2015中编译后使用。(注意:pycocotools编译对VS版本有要求,VS2013以下或不成功)
5 匹配keras版本
匹配前运行会报错:
`AttributeError: module 'keras.engine.topology' has no attribute 'load_weights_from_hdf5_group_by_name'`
原因是版本不匹配。在安装mask_rcnn时,requirements.txt要求的是keras>=2.0.8,默认安装的是keras2.2.2,但是load_weights_from_hdf5_group_by_name只存在于keras2.0.8。
解决方法:
pip uninstall keras
pip install keras==2.0.8
注意:
模型mask_rcnn_coco.h5可能下载不成功,在第一个模块执行时报错:
OSError: Unable to open file (truncated file: eof = 102137856, sblock->base_addr = 0, stored_eof = 257557808)
删除模型文件mask_rcnn_coco.h5,重新下载即可。
参考及链接:
pycocotools包网盘链接:
https://pan.baidu.com/s/1AVyld-kCdujuLwBilhbY8w
提取码:w4qi