4 个月前

WeakSAM:弱监督实例级识别与分割一切相遇

WeakSAM:弱监督实例级识别与分割一切相遇

摘要

使用不精确监督进行弱监督视觉识别是一个关键但具有挑战性的学习问题。它显著降低了人工标注成本,并且传统上依赖于多实例学习和伪标签方法。本文介绍了WeakSAM,该方法通过利用预训练的视觉基础模型(即Segment Anything Model,SAM)中包含的世界知识,解决了弱监督目标检测(WSOD)和分割问题。WeakSAM通过自适应伪地面真值(PGT)生成和感兴趣区域(RoI)丢弃正则化,克服了传统WSOD再训练中的两个主要限制,即伪地面真值的不完整性以及噪声PGT实例。此外,WeakSAM还解决了SAM在自动目标检测和分割中需要提示和类别无感知的问题。我们的实验结果表明,WeakSAM在WSOD和WSIS基准测试中大幅超越了以往的最先进方法,分别平均提高了7.4%和8.5%。代码可在以下网址获取:\url{https://github.com/hustvl/WeakSAM}。

代码仓库

hustvl/weaksam
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
image-level-supervised-instance-segmentationWeakSAM-Mask RCNN (with SAM)
mAP@0.25: 70.3
mAP@0.5: 59.6
mAP@0.7: 43.1
mAP@0.75: 36.2
image-level-supervised-instance-segmentationWeakSAM-Mask2Former (with SAM)
mAP@0.25: 73.4
mAP@0.5: 64.4
mAP@0.7: 49.7
mAP@0.75: 45.3
image-level-supervised-instance-segmentation-1WeakSAM-Mask RCNN (with SAM)
AP: 21.0
AP@50: 34.5
AP@75: 22.2
image-level-supervised-instance-segmentation-1WeakSAM-Mask2Former (with SAM)
AP: 25.9
AP@50: 39.9
AP@75: 27.9
image-level-supervised-instance-segmentation-2WeakSAM-Mask RCNN (with SAM)
AP: 20.6
AP@50: 33.9
AP@75: 22.0
image-level-supervised-instance-segmentation-2WeakSAM-Mask2Former (with SAM)
AP: 25.2
AP@50: 38.4
AP@75: 27.0
weakly-supervised-object-detection-on-ms-cocoWeakSAM-OICR-DINO (with SAM)
AP: 24.9
weakly-supervised-object-detection-on-ms-cocoWeakSAM-MIST-Faster RCNN (with SAM)
AP: 23.8
weakly-supervised-object-detection-on-ms-cocoWeakSAM-MIST-DINO (with SAM)
AP: 26.6
weakly-supervised-object-detection-on-ms-cocoWeakSAM-MIST (with SAM)
AP: 22.9
weakly-supervised-object-detection-on-ms-cocoWeakSAM-OICR (with SAM)
AP: 19.9
weakly-supervised-object-detection-on-ms-cocoWeakSAM-OICR-Faster RCNN (with SAM)
AP: 22.3
weakly-supervised-object-detection-on-pascalWeakSAM-OICR-DINO (with SAM)
MAP: 63.7
weakly-supervised-object-detection-on-pascalWeakSAM-MIST-DINO (with SAM)
MAP: 70.2
weakly-supervised-object-detection-on-pascalWeakSAM-MIST-Faster RCNN (with SAM)
MAP: 69.2
weakly-supervised-object-detection-on-pascalWeakSAM-MIST (with SAM)
MAP: 66.9
weakly-supervised-object-detection-on-pascalWeakSAM-OICR-Faster RCNN (with SAM)
MAP: 62.9
weakly-supervised-object-detection-on-pascalWeakSAM-OICR (with SAM)
MAP: 58.4
weakly-supervised-object-detection-on-pascal-1WeakSAM-OICR (with SAM)
MAP: 58.9
weakly-supervised-object-detection-on-pascal-1WeakSAM-MIST-Faster RCNN (with SAM)
MAP: 71.8
weakly-supervised-object-detection-on-pascal-1WeakSAM-MIST (with SAM)
MAP: 67.4
weakly-supervised-object-detection-on-pascal-1WeakSAM-MIST-DINO (with SAM)
MAP: 73.4
weakly-supervised-object-detection-on-pascal-1WeakSAM-OICR-Faster RCNN (with SAM)
MAP: 65.7
weakly-supervised-object-detection-on-pascal-1WeakSAM-OICR-DINO (with SAM)
MAP: 66.1

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
WeakSAM:弱监督实例级识别与分割一切相遇 | 论文 | HyperAI超神经