其他分享
首页 > 其他分享> > Mask RCNN里的balloon跑通例子

Mask RCNN里的balloon跑通例子

作者:互联网

Mask RCNN里的balloon跑通例子

1.win10下CUDA和CUDNN的安装
2.Mask RCNN环境部署
3.mask_rcnn_balloon.h5balloon数据集的下载
配置好以上环境后,就可以开始训练了,以下均以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 RCNN终于跑通了,不容易啊!
下一步,开始认真看代码了!

标签:--,mask,balloon,跑通,Mask,h5,RCNN
来源: https://blog.csdn.net/chengxuyuanliwanwan/article/details/111566219