首页 新闻 赞助 找找看

mask-RCNN模型训练发现mask存在重叠情况

0
[已解决问题] 解决于 2020-12-18 10:08

请教下各位大神·~~
我用mask-RCNN训练自己的数据集,发现模型预测的时候,存在一个问题:
一张图片中如有多个相同类别,其识别的时候,会区分为不同的物体,名称也都是类别1.。。。但是同一个物体会被重复多次识别。。。这个是怎么回事??
截图如下

老笨啊的主页 老笨啊 | 初学一级 | 园豆:6
提问于:2019-12-10 09:57
< >
分享
最佳答案
0

可以做非极大值抑制吧

奖励园豆:5
yytxdy | 小虾三级 |园豆:1680 | 2019-12-10 15:42

具体该如何做呢??
我是直接用的mask-RCNN模型做迁移学习,我很奇怪这个不是应该内部设置了NMS么?
还是说我在标注的时候,需要主要在一张图中区分同一类别的不同物体,比如在一张图中,标注类别1-1,类别1-2,类别1-3等这样的标记??

老笨啊 | 园豆:6 (初学一级) | 2019-12-11 11:24
清除回答草稿
   您需要登录以后才能回答,未注册用户请先注册