本文详细介绍了如何在GPU环境下配置和使用基于Tensorflow和Keras的Mask RCNN。首先安装Anaconda，然后分别安装Tensorflow-GPU和Keras。接着安装相关依赖库，下载代码和预训练权重文件mask_rcnn_coco.h5。为解决pycocotools问题，从特定源下载并编译。再下载COCO数据集，包括annotations、train data和val data。最后通过Jupyter Notebook运行demo验证配置成功。
这篇主要是讲如何配置Mask RCNN(基于tensorflow + keras的gpu版本), github地址： https://github.com/matterport/Mask_RCNN
Step 1 : 安装anaconda
从官网https://www.anaconda.com/download/上下载适合自己的版本，如果嫌速度慢，可以从该网址下载：Anaconda清华镜像 我下载的是Anaconda3-5.0.1-Linux-x86_64.sh
Step 2 : 安装tensorflow：
对于CPU版本：pip install tensorflow
对于GPU版本：pip install tensorflow-gpu
Step 3 :安装keras
pip install keras安装完tensorflow和keras之后，自己测试一下是否安装成功。接下来按照github上的README进行安装：
Step 4 :安装依赖库
pip3 install -r requirements.txt
最低0.47元/天 解锁文章
扫一扫
2038
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
