Fine-tuning SAM 2 to Output Segmentation Masks with Class Names

facebookresearch / sam2

The repository provides code for running inference with the Meta Segment Anything Model 2 (SAM 2), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.

Apache License 2.0

12.85k stars 1.21k forks source link

Fine-tuning SAM 2 to Output Segmentation Masks with Class Names #326

Open jeet-patel-us opened 2 months ago

jeet-patel-us commented 2 months ago

I am working on a project that involves using SAM 2 for image segmentation, and I would like to explore the possibility of fine-tuning the model to output segmentation masks along with their corresponding class names. The goal is to not only obtain the segmentation masks but also associate each segmented object with its respective class name.

Shiyao-Xu commented 1 month ago

Hey, same question here, did you manage to solve this task?