flaviagiammarino
commited on
Commit
•
4753789
1
Parent(s):
901b402
Update README.md
Browse files
README.md
CHANGED
@@ -10,9 +10,10 @@ MedSAM is a fine-tuned version of [SAM](https://huggingface.co/docs/transformers
|
|
10 |
|
11 |
## Model Description
|
12 |
|
13 |
-
MedSAM was trained on a large-scale medical image segmentation dataset
|
|
|
14 |
|
15 |
-
MedSAM was initialized with the pre-trained SAM model with the ViT-Base backbone. The prompt encoder
|
16 |
The training was performed for 100 epochs with a batch size of 160 using the AdamW optimizer with a learning rate of 10−4 and a weight decay of 0.01.
|
17 |
|
18 |
- **Repository:** [MedSAM Official GitHub Repository](https://github.com/bowang-lab/medsam)
|
|
|
10 |
|
11 |
## Model Description
|
12 |
|
13 |
+
MedSAM was trained on a large-scale medical image segmentation dataset containing 1,090,486 medical image-mask pairs collected from different publicly available sources.
|
14 |
+
The image-mask pairs cover 15 different imaging modalities and over 30 cancer types.
|
15 |
|
16 |
+
MedSAM was initialized with the pre-trained SAM model with the ViT-Base backbone. The prompt encoder weights were frozen, while the image encoder and mask decoder weights were updated during training.
|
17 |
The training was performed for 100 epochs with a batch size of 160 using the AdamW optimizer with a learning rate of 10−4 and a weight decay of 0.01.
|
18 |
|
19 |
- **Repository:** [MedSAM Official GitHub Repository](https://github.com/bowang-lab/medsam)
|