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

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

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

人工智能图像识别目标检测 Mask-RCNN

老笨啊 | 初学一级 | 园豆：6
提问于：2019-12-10 09:57

< >

最佳答案

可以做非极大值抑制吧

奖励园豆：5

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

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

老笨啊 | 园豆：6 (初学一级) | 2019-12-11 11:24

清除回答草稿

您需要登录以后才能回答，未注册用户请先注册。

欢迎，请先 登录 或者 注册 。