WeakSAM: Segment Anything Meets Weakly-supervised Instance-level Recognition

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Zhu, Lianghui, Zhou, Junwei, Liu, Yan, Hao, Xin, Liu, Wenyu, Wang, Xinggang
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866929462334980096
author Zhu, Lianghui
Zhou, Junwei
Liu, Yan
Hao, Xin
Liu, Wenyu
Wang, Xinggang
author_facet Zhu, Lianghui
Zhou, Junwei
Liu, Yan
Hao, Xin
Liu, Wenyu
Wang, Xinggang
contents Weakly supervised visual recognition using inexact supervision is a critical yet challenging learning problem. It significantly reduces human labeling costs and traditionally relies on multi-instance learning and pseudo-labeling. This paper introduces WeakSAM and solves the weakly-supervised object detection (WSOD) and segmentation by utilizing the pre-learned world knowledge contained in a vision foundation model, i.e., the Segment Anything Model (SAM). WeakSAM addresses two critical limitations in traditional WSOD retraining, i.e., pseudo ground truth (PGT) incompleteness and noisy PGT instances, through adaptive PGT generation and Region of Interest (RoI) drop regularization. It also addresses the SAM's problems of requiring prompts and category unawareness for automatic object detection and segmentation. Our results indicate that WeakSAM significantly surpasses previous state-of-the-art methods in WSOD and WSIS benchmarks with large margins, i.e. average improvements of 7.4% and 8.5%, respectively. The code is available at \url{https://github.com/hustvl/WeakSAM}.
format Preprint
id arxiv_https___arxiv_org_abs_2402_14812
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle WeakSAM: Segment Anything Meets Weakly-supervised Instance-level Recognition
Zhu, Lianghui
Zhou, Junwei
Liu, Yan
Hao, Xin
Liu, Wenyu
Wang, Xinggang
Computer Vision and Pattern Recognition
Artificial Intelligence
Weakly supervised visual recognition using inexact supervision is a critical yet challenging learning problem. It significantly reduces human labeling costs and traditionally relies on multi-instance learning and pseudo-labeling. This paper introduces WeakSAM and solves the weakly-supervised object detection (WSOD) and segmentation by utilizing the pre-learned world knowledge contained in a vision foundation model, i.e., the Segment Anything Model (SAM). WeakSAM addresses two critical limitations in traditional WSOD retraining, i.e., pseudo ground truth (PGT) incompleteness and noisy PGT instances, through adaptive PGT generation and Region of Interest (RoI) drop regularization. It also addresses the SAM's problems of requiring prompts and category unawareness for automatic object detection and segmentation. Our results indicate that WeakSAM significantly surpasses previous state-of-the-art methods in WSOD and WSIS benchmarks with large margins, i.e. average improvements of 7.4% and 8.5%, respectively. The code is available at \url{https://github.com/hustvl/WeakSAM}.
title WeakSAM: Segment Anything Meets Weakly-supervised Instance-level Recognition
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2402.14812