win10 64位配置mask-rcnn环境的坑(主要是安装imgaug和pycocotools,解决“No module named ‘pycocotools._mask”)

python3.5.4, tensorflow-gpu版本1.3(cuda8.0,cudnn6.0), keras版本2.0.8。
mask-rcnn源码地址https://github.com/csq20081052/Mask_RCNN,文件夹下的samples/demo.ipynb可以在jupyter notebook中直接运行。
在win10系统上运行该demo时候有两个大坑:imgaug和pycocotools的安装
1、imgaug的安装
虚拟环境下直接pip install imgaug应该会报错(顺利的话最好)
解决办法:下载https://github.com/aleju/imgaug中的文件,将其中的imgaug文件夹直接复制到虚拟环境下的Lib/site-packages目录下即可。
2、pycocotools的安装
巨坑,直接pip install pycocotools会报错,下载https://github.com/philferriere/cocoapi文件,虚拟环境下cd 到PythonAPI目录下,执行
python setup.py build_ext install
会编译并将pycocotools安装到虚拟环境下的Lib/site-packages目录中,编译需要安装visual studio2015。(有的博主推荐的https://github.com/cocodataset/cocoapi地址,我在这个文件中执行python setup.py build_ext instal会报bug)
安装完成之后python命令进入python环境,能运行from pycocotools.coco import COCO和import pycocotools._mask as _mask即大功告成。
PS:如果demo.ipynb的当前目录中有pycocotools文件夹则import pycocotools的时候会优先import当前目录下的文件,如果在命令行中import没有问题在jupyter notebook上发现“No module named ‘pycocotools._mask”等问题,则只需把虚拟环境中的pycocotools文件夹替换掉当前目录下的pycocotools文件夹

你可能感兴趣的:(win10 64位配置mask-rcnn环境的坑(主要是安装imgaug和pycocotools,解决“No module named ‘pycocotools._mask”))