ubantu18.04下基于cuda11.0的MaskRCNN环境搭建

介绍一下我的电脑是显卡3080ti，好像只能支持cuda11.0以上。自己摸索最后通过了balloon气球的实验。记录一下。下面的俩个外部环境都可以
cuda11.1 cudnn是8.0.5
cuda11.2 cudnn8.1.0

1.anaconda安装

自己在网上百度一下，不再赘述。

2.tensorflow-gpu keras安装

cuda=11.0 cudnn=8.0.5 支持的tensorflow-gpu是2.4 tensorflow-gpu支持的keras=2.4.3.不过在tensorflow2以后好像keras的一部分功能都已经融入tensorflow了。tf.keras.xxxx 这个我们不管继续我们的工作

tensorflow cudnn cudatoolkit对应图如下

原址https://tensorflow.google.cn/ 的安装界面下，选择linux 划到最下面

tensorflow和keras对应关系

下载maskrcnn源代码

项目源代码地址：https://github.com/matterport/Mask_RCNN

git clone https://github.com/matterport/Mask_RCNN
#下载如果很慢 ctrl+z停止后 把https换成git 再试一下很快的

还是不能解决的话，MASKRCNN_gihub中国镜像
1.你自己下载完成，拷贝复制
2.进去以后换一下https的clone的地址
下文中所有的根目录就是你这个文件夹进去的页面（Mask_RCNN-master）

开始搭建环境

conda create -n mask python=3.6
source activate mask #激活环境
##为了下载的快一点 你最好在激活环境以后换一下镜像源
#gedit ~/.condarc
#下面的复制进去 把#号去了 
#channels:
#  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
#  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
# - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
#  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
#ssl_verify: true
#保存退出
conda install cudatoolkit==11.0
conda install cudnn==8.0.5
pip install tensorflow-gpu=2.4 -i https://pypi.douban.com/simple/
pip install keras=2.4.3 -i https://pypi.douban.com/simple/

以上的如果你用conda pip命令装不好
1、确保你已经换源了
2、你用conda search xxx查一下源里面有哪个版本
例如我cudnn=8.0.4 清华源就没有 查询以后有8.0.5我就装的8.0.5. 只要前俩位正确就行的。
主要的搭建完了，剩下的打开MASKRCNN目录下的requirements.txt 用
pip install xxxx -i https://pypi.douban.com/simple/ 安装一下

在maskrcnn根目录下

python3 setup.py install

这个运行以后，可能在后续的报错中。代码错误显示在anaconda/envs/环境名字/…/mask_rcnn_xxx.rgg(这个是压缩包)
解决方法：你把这个删除了，报错位置就会是maskrcnn文件根目录。

下载权重文件

网址：https://github.com/matterport/Mask_RCNN/releases
里面有2.1和2.0 我是都下载了但是只用到了2.1的。
2.1的是balloon实验数据和权重文件及其源代码。我下载了Assets下的前三个。

ballon_data你在mask根目录下建立一个datasets文件下把balloon整个解压进去。mask_rcnn_balloon.h5文件放在根目录下

pycocotools的安装

git clone https://github.com/cocodataset/cocoapi.git

如果不成功，pycocotools中国镜像下载拷贝进去。
这个文件放在mask_rcnn根目录下。
进入coco-mater的PythonAPI文件夹下，打开终端

#激活创建的虚拟环境
source activate mask 
make
make install
python setup.py install

把生成的pycocotools拷贝到根目录下。conda list看有没有pycococtools。

运行demo

我用的pycharm
打开maskrcnn文件夹，在samples/balloon文件夹下运行你看一下readme
python3 balloon.py splash --weights=权重文件的绝对路径（带.h5）–image=图片路径（带.图片格式的）

运行中的错误

改错地址
改错地址1
如果出现：tensorflow xxxxxx NotFoundError: No algorithm worked!
没有保存预测图片
解决方法：

from tensorflow.compat.v1 import ConfigProto
from tensorflow.compat.v1 import InteractiveSession
 
config = ConfigProto()
config.gpu_options.allow_growth = True
session = InteractiveSession(config=config)

贴在前面再跑一次，就可以了。
后续的文件，下个月我上传百度云，贴在评论里。