The Segment Anything Model (SAM) produces high-quality object masks from input prompts such as points or boxes,
and it can be used to generate masks for all objects in an image. It has been trained on a dataset of 11 million
images and 1.1 billion masks and has strong zero-shot performance on a variety of segmentation tasks.
https://github.com/facebookresearch/segment-anything
1device = "cuda" if torch.cuda.is_available() else "cpu"
2processor = SamProcessor.from_pretrained("wanglab/medsam-vit-base")
3model = SamModel.from_pretrained("ayoubkirouane/Breast-Cancer_SAM_v1").to(device)