4 个月前

BoxInst: 基于框标注的高性能实例分割

BoxInst: 基于框标注的高性能实例分割

摘要

我们提出了一种高性能的方法,仅使用边界框注释即可实现像素级实例分割。尽管这一设置在文献中已有研究,但在这里我们展示了显著更强的性能,通过简单的设计(例如,在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-onBoxInst
AP_50: 61.4
AP_75: 37.0
mask AP: 36.5
box-supervised-instance-segmentation-on-cocoBoxInst
mask AP: 35.0
weakly-supervised-instance-segmentation-on-2BoxInst (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-2BoxInst (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-2BoxInst (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-2BoxInst (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

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
BoxInst: 基于框标注的高性能实例分割 | 论文 | HyperAI超神经