一、配置环境:
Mask R-CNN是基于Python3,Keras,TensorFlow。
Python 3.4+
TensorFlow 1.3+
Keras 2.0.8+
Jupyter Notebook
Numpy, skimage, scipy, Pillow, cython, h5py
二、安装
1)安装依赖关系
pip3 install -r requirements.txt )
2)克隆这个存储库
3)从存储库根目录运行安装程序
python3 setup.py install
4)下载好训练好的权重mask_rcnn_coco.h5
5)对于pycocotools库安装方法如下
git clone https://github.com/pdollar/coco
cd coco/PythonAPI
将makefile中的python 改为python3
然后先运行安装python3-dev
然后命令行输入
make -j8
然后将pycocotools文件夹复制到mask-rcnn下
最 后再
sudo pip3 install h5py
coco数据集:
http://images.cocodataset.org/zips/train2014.zip
http://images.cocodataset.org/zips/val2014.zip
下载对应的json文件
https://dl.dropboxusercontent.com/s/o43o90bna78omob/instances_minival2014.json.zip?dl=0
https://dl.dropboxusercontent.com/s/s3tw5zcg7395368/instances_valminusminival2014.json.zip?dl=0