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!