Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Self-Prompting Polyp Segmentation in Colonoscopy using Hybrid Yolo-SAM 2 Model

About

Early diagnosis and treatment of polyps during colonoscopy are essential for reducing the incidence and mortality of Colorectal Cancer (CRC). However, the variability in polyp characteristics and the presence of artifacts in colonoscopy images and videos pose significant challenges for accurate and efficient polyp detection and segmentation. This paper presents a novel approach to polyp segmentation by integrating the Segment Anything Model (SAM 2) with the YOLOv8 model. Our method leverages YOLOv8's bounding box predictions to autonomously generate input prompts for SAM 2, thereby reducing the need for manual annotations. We conducted exhaustive tests on five benchmark colonoscopy image datasets and two colonoscopy video datasets, demonstrating that our method exceeds state-of-the-art models in both image and video segmentation tasks. Notably, our approach achieves high segmentation accuracy using only bounding box annotations, significantly reducing annotation time and effort. This advancement holds promise for enhancing the efficiency and scalability of polyp detection in clinical settings https://github.com/sajjad-sh33/YOLO_SAM2.

Mobina Mansoori, Sajjad Shahabodini, Jamshid Abouei, Konstantinos N. Plataniotis, Arash Mohammadi• 2024

Related benchmarks

TaskDatasetResultRank
Polyp SegmentationETIS
Dice Score94.9
108
Polyp SegmentationCVC-ClinicDB
Dice Coefficient95.1
81
Polyp SegmentationCVC-ColonDB
mDice91.8
66
Video Polyp SegmentationSUN-SEG Hard (Unseen)
S Alpha Score89.4
27
Video Polyp SegmentationSUN-SEG Easy (Unseen)
S-alpha Score0.9
27
Polyp SegmentationKvasir-Seg
mDice0.866
27
Video Polyp SegmentationSUN-SEG Easy (Seen)
S_alpha93.7
21
Polyp SegmentationCVC300
mDice88.9
19
Polyp Video SegmentationSUN-SEG Seen-Hard
S_alpha89.3
12
Video Polyp SegmentationPolypgen 23 sequence videos
mDice80.8
9
Showing 10 of 10 rows

Other info

Code

Follow for update