Mask RCNN 学习笔记(1) linux下 配置安装
系统参数: 处理器 Intel(R) Core(TM) i7-8700K CPU @ 3.70GHz
内存: 65836032KB
显卡 : 01:00.0 VGA compatible controller: NVIDIA Corporation GP102 [GeForce GTX 1080 Ti] (rev ff)
03:00.0 VGA compatible controller: NVIDIA Corporation GP102 [GeForce GTX 1080 Ti] (rev ff)
安装步骤:
Github上源码地址: https://github.com/matterport/Mask_RCNN
按照官网给出的安装安装步骤 一步一步来:
Installation
-
Install dependencies
pip3 install -r requirements.txt
Clone this repository
-
在进行下一步之前,先操作这3步:如果你的Keras版本高于2.08 卸载 1) keras : pip uninstall keras 2)安装2.0.8版本的keras:pip install keras==2.0.8 3) 打开 mrcnn/model.py, 把"topology" 替换为"saving" 参考地址: https://github.com/matterport/Mask_RCNN/issues/694
这样做是为了防止报错:
module 'keras.engine.topology' has no attribute 'load_weights_from_hdf5_group_by_name'
-
Run setup from the repository root directory
python3 setup.py install
下载 预先训练好的COCO模型权重 mask_rcnn_coco.h5。
原始链接: https://github.com/matterport/Mask_RCNN/releases
国内不好下载的话从百度云上下载: https://pan.baidu.com/s/1q41lxiuDKjUsN2ZvUagHWA
6 安装 pycocotools
跟官网说的不大一样 这个好像是必须安装的 因为给出的demo里面需要这个。
- Linux: https://github.com/waleedka/coco
- Windows: https://github.com/philferriere/cocoapi. You must have the Visual C++ 2015 build tools on your path (see the repo for additional details)
最后运行 sample里面的demo ,如下图:
有问题欢迎交流 QQ: 865883265