环境
Ubuntu20.04
2080ti
cuda11.3
tensorflow-gpu2.6.0
keras2.1.6
安装
- cuda+cudnn参考 安装 CUDA11.3
- Ubuntu Anaconda 虚拟环境安装tensorflow-gpu==2.6.0
- 安装keras:pip3 install keras==2.1.6
下载代码、运行
- 下载修改的MaskRcnn代码:https://github.com/leekunhee/Mask_RCNN
- 删除requirements.txt中的tensorflow1.13.1和
keras2.0.8两行 - 执行pip install -r requirements.txt
- 执行:python setup.py install
- 下载balloon数据集,coco权重参数到Mask_Rcnn-master的根目录。链接:https://github.com/matterport/Mask_RCNN/releases
Mask R-CNN 2.1的balloon_dataset.zip
Mask R-CNN 2.0的mask_rcnn_coco.h5
目录结构如下:
- 进入sample/balloon/中运行python balloon.py train --dataset=…/…/balloon --weight=coco
于是OK!!!!!