May, 2024

竞争像素:一种弱监督分割的自对弈算法

TL;DR基于图像级标签的弱监督分割方法缺乏标签和感兴趣区域之间的明确对应关系,为此,我们提出了一种基于强化学习自我博弈的新型弱监督分割方法,通过将分割视为两个代理竞争选择包含感兴趣区域的图像块,从而实现对目标的定位。通过使用仅基于图像级二分类标签的目标存在检测器进行预训练,我们在每个时间步计算分数以计算代理训练的奖励,该分数表示选择中物体存在的可能性。此外,我们提出了一种游戏终止条件,当耗尽所有包含感兴趣区域的图像块后,任一方可以调用该条件,然后从中选择最终的图像块。通过这种竞争模式,我们确保最小化过度或不足分割的问题。在四个数据集上进行的广泛实验表明,与最新的最先进方法相比,我们的方法有显著的性能提升。