ayoubkirouane commited on
Commit
b2d31d9
·
1 Parent(s): fc4b6f8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +37 -0
README.md CHANGED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ library_name: transformers
6
+ pipeline_tag: image-segmentation
7
+ tags:
8
+ - medical
9
+ - segmentation
10
+ datasets:
11
+ - nielsr/breast-cancer
12
+ ---
13
+
14
+ ## Description :
15
+
16
+ **Breast cancer segmentation** is the task of identifying and segmenting the breast tumor region in **medical images**,
17
+ such as mammograms and ultrasound images. This is an important task in the diagnosis and treatment of breast cancer,
18
+ as it helps clinicians to better understand the extent of the disease and plan appropriate interventions.
19
+
20
+ **The Segment Anything Model (SAM)** is a state-of-the-art deep learning model for image segmentation.
21
+ SAM is a vision transformer-based model that has been shown to achieve excellent performance on a
22
+ variety of natural image segmentation tasks.
23
+
24
+ ## Base Model:
25
+
26
+ **The Segment Anything Model (SAM)** produces high-quality object masks from input prompts such as points or boxes,
27
+ and it can be used to generate masks for all objects in an image. It has been trained on a dataset of 11 million
28
+ images and 1.1 billion masks and has strong zero-shot performance on a variety of segmentation tasks.
29
+ **https://github.com/facebookresearch/segment-anything**
30
+
31
+
32
+ ## Get Started with the Model
33
+ ``` python
34
+ device = "cuda" if torch.cuda.is_available() else "cpu"
35
+ processor = SamProcessor.from_pretrained("wanglab/medsam-vit-base")
36
+ model = SamModel.from_pretrained("ayoubkirouane/Breast-Cancer_SAM_v1").to(device)
37
+ ```