我在我的项目中使用Mask_RCNN来分割设备的各个部分。它工作正常,除了输出(分段图)具有重叠的区域这一事实之外,这在稍后的管道中会带来巨大的问题。请参见下图以供参考。例如,该圆形(橙色的磁盘)与读取器(紫色)重叠。
换句话说,我想拥有没有重叠线段的分割图。有人在使用Mask_RCNN进行细分任务吗?由于它基于区域建议而工作,因此我对如何解决该问题感到困惑。我们如何确保每个检测到的实例都有其自身的像素,而不是其他实例?我们如何使这些细分市场独一无二?
# Compute overlaps [rpn_rois, gt_boxes]
overlaps = np.zeros((rpn_rois.shape[0], gt_boxes.shape[0]))
for i in range(overlaps.shape[1]):
gt = gt_boxes[i]
overlaps[:, i] = utils.compute_iou(
gt, rpn_rois, gt_box_area[i], rpn_roi_area)
我查看了code,发现甚至有评论提到重叠。但是,我无法掌握应该在其中进行哪些修改以避免重叠。不能只是删除整个代码段。
有人可以在这方面帮助我吗?
P.S:我确实在那里的问题中问过,但是没有得到答复。