win10 运行Mask_RCNN

923 阅读1分钟

Mask_RCNN github 项目似乎没有考虑windows运行的兼容问题,按照以下步骤可运行demo

安装CUDA9.0 CUDA9.0.176下载
提取码:v3g9
安装CUDA时勾选将CUDA安装目录添加到系统环境变量path
安装cudnn7.5 cudnn7.5.0.56下载地址
提取码:pq4g
将cudnn解压后目录的文件拷贝到CUDA安装路径下对应目录
测试cudnn: nvcc -V

conda新建虚拟环境

conda create -n tf python==3.6

pip安装requirements中的包
numpy
scipy
Pillow
cython
matplotlib
scikit-image
tensorflow>=1.3.0 #pip install tensorflow-gpu==1.9.0
keras>=2.0.8
opencv-python
h5py
imgaug
IPython[all]\

安装tensorflow时默认安装的protobuf需要回退到3.6.0

pip install protobuf-3.6.0

安装COCO数据集

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

运行demo:

jupyter notebook demo.ipydb

第一个section会先下载模型mask_rcnn_coco.h5 文件245M比较大,下不动的同学可以点这个手动下载
mask_rcnn_coco.h5
提取码:i7ud
将下好的文件放在项目根目录即可

运行成功

happy coding!