标签:-- mask balloon 跑通 Mask h5 RCNN
Mask RCNN里的balloon跑通例子
1.win10下CUDA和CUDNN的安装
2.Mask RCNN环境部署
3.mask_rcnn_balloon.h5和balloon数据集的下载
配置好以上环境后,就可以开始训练了,以下均以balloon数据集为基础来讲解。
- 首先,修改balloon.py文件里的Mask RCNN在自己电脑的路径,如下图:
- 打开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
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。