flaviagiammarino commited on
Commit
4753789
1 Parent(s): 901b402

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
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 with 1,090,486 medical image-mask pairs collected from different publicly available sources including 15 imaging modalities and over 30 cancer types.
 
14
 
15
- MedSAM was initialized with the pre-trained SAM model with the ViT-Base backbone. The prompt encoder's weights were frozen, while all trainable parameters in the image encoder and mask decoder were updated during training.
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)