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文件夹

评论 15
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值