Mask RCNN里的balloon跑通例子

1.win10下CUDA和CUDNN的安装

2.Mask RCNN环境部署

3.mask_rcnn_balloon.h5和balloon数据集的下载

配置好以上环境后，就可以开始训练了，以下均以balloon数据集为基础来讲解。

1. 首先，修改balloon.py文件里的Mask RCNN在自己电脑的路径，如下图：
   
2. 打开terminal，将路径转到samples的balloon目录下，输入以下命令，开始训练：

python balloon.py train --dataset="D:/Mask_RCNN-master/datasets/balloon" --weights="D:/Mask_RCNN-master/mask_rcnn_balloon.h5"

训练过程如下：

大概训练1个小时多，就跑完了，会生成logs目录，该目录下生成很多.h5文件，这里只截取部分展示：

3. 接着使用生成的.h5文件进行测试，这里使用mask_rcnn_balloon_0002.h5，输入以下命令：

python balloon.py splash --weights="D:/Mask_RCNN-master/logs/balloon20201222T1742/mask_rcnn_balloon_0002.h5" --image="410488422_5f8991f26e_b.jpg"

结果如图：

测试原图：

测试生成图：

可以看到，成功实现了使用生成的蒙版来保持气球的颜色，同时将图像的其余部分更改为灰度。
至此，Mask R-CNN终于跑通了，不容易啊！
下一步，开始认真看代码了！