Automatic Prompt Generation for Zero-Shot Single Object Frame Segmentation in Videos Using Classification Models: A Polyp Case Study
Video object segmentation is vital for applications like medical diagnostics, but acquiring dense pixel-level annotations, especially for specialized domains like polyp segmentation, remains a major bottleneck. Foundational models offer zero-shot segmentation but typically require manual prompting, which is impractical for long videos. We propose Map2VidSeg, a novel pipeline that automatically generates prompts from imagelevel classification labels. It leverages localization cues (attention maps/CAMs) from a trained image classifier (ViT/CNN) to create bounding box prompts. These guide an efficient model (YOLOE) with tracking (BOT-SORT) and bidirectional propagation for initial segmentation. Optionally, a high-fidelity model (SAM-2) refines these masks using temporal memory and fusion. Demonstrated on the challenging SUN-SEG benchmark, finetuned DINOv2 (ViT) prompts significantly outperform DenseNet-121 (CNN). Our best configuration (DINOv2+YOLOE+SAM-2 Bidirectional) achieves Dice/mIoU 0.76/0.70 (Easy Unseen) and 0.66/0.60 (Hard Unseen), showcasing the viability of robust video segmentation without segmentation training data.
Authors
- Hanna Borgli (ORCID: https://orcid.org/0000-0001-9925-6134)
- Pål Halvorsen (ORCID: https://orcid.org/0000-0003-2073-7029)
- Michael A. Riegler (ORCID: https://orcid.org/0000-0002-3153-2064)
- Håkon Kvale Stensland (ORCID: https://orcid.org/0000-0002-1085-8540)
Institutions
- University of Oslo (NO)
- Simula Research Laboratory (NO)
- Simula Metropolitan Center for Digital Engineering (NO)
Publication Details
- Published
- 2025-06-18
- DOI
- https://doi.org/10.1109/cbms65348.2025.00018
- Primary Topic
- Video Analysis and Summarization
- Type
- article
- Field-Weighted Citation Impact
- 0.00