
摘要
我们提出了一种高性能的方法,仅使用边界框注释即可实现像素级实例分割。尽管这一设置在文献中已有研究,但在这里我们展示了显著更强的性能,通过简单的设计(例如,在COCO数据集上将Hsu等人(2019)先前报告的最佳掩码AP从21.1%大幅提升至31.6%)。我们的核心思想是在不修改分割网络本身的情况下,重新设计实例分割中学习掩码的损失函数。新的损失函数可以在不依赖掩码注释的情况下监督掩码训练。这通过两个损失项得以实现,即:1)一个代理项,最小化真实边界框投影与预测掩码之间的差异;2)一个成对损失项,利用邻近像素颜色相似则很可能具有相同类别标签的先验知识。实验结果表明,重新设计的掩码损失函数仅使用边界框注释就能生成高质量的实例掩码。例如,在不使用任何掩码注释的情况下,采用ResNet-101骨干网络和3倍训练计划,我们在COCO测试开发集上达到了33.2%的掩码AP(而完全监督方法的对应值为39.1%)。我们在COCO和Pascal VOC上的优秀实验结果表明,该方法显著缩小了弱监督和全监督实例分割之间的性能差距。代码可从以下地址获取:https://git.io/AdelaiDet
代码仓库
aim-uofa/adet
pytorch
GitHub 中提及
aim-uofa/AdelaiDet
pytorch
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| box-supervised-instance-segmentation-on | BoxInst | AP_50: 61.4 AP_75: 37.0 mask AP: 36.5 |
| box-supervised-instance-segmentation-on-coco | BoxInst | mask AP: 35.0 |
| weakly-supervised-instance-segmentation-on-2 | BoxInst (ResNet-50-FPN) | AP: 32.1 AP@50: 55.1 AP@75: 32.4 AP@L: 43.5 AP@M: 34.3 AP@S: 15.6 |
| weakly-supervised-instance-segmentation-on-2 | BoxInst (ResNet-101-BiFPN) | AP: 33.9 AP@50: 57.7 AP@75: 34.5 AP@L: 46.6 AP@M: 36.1 AP@S: 16.5 |
| weakly-supervised-instance-segmentation-on-2 | BoxInst (ResNet-101-FPN) | AP: 33.2 AP@50: 56.5 AP@75: 33.6 AP@L: 45.1 AP@M: 35.3 AP@S: 16.2 |
| weakly-supervised-instance-segmentation-on-2 | BoxInst (ResNet-101-DCN-BiFPN) | AP: 35.0 AP@50: 59.3 AP@75: 35.6 AP@L: 48.9 AP@M: 37.2 AP@S: 17.1 |