flaviagiammarino
/

medsam-vit-base

Mask Generation

Inference Endpoints

Model card Files Files and versions Community

flaviagiammarino commited on Jul 11, 2023

Commit

78716a6

•

1 Parent(s): 2095a63

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -11,7 +11,7 @@ MedSAM is a fine-tuned version of [SAM](https://huggingface.co/docs/transformers
 ## Model Description
 MedSAM was trained on a large-scale medical image segmentation dataset of 1,090,486 medical image-mask pairs collected from different publicly available sources.
-The image-mask pairs cover 15 different imaging modalities and over 30 cancer types.
 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.
 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.

 ## Model Description
 MedSAM was trained on a large-scale medical image segmentation dataset of 1,090,486 medical image-mask pairs collected from different publicly available sources.
+The image-mask pairs cover 15 imaging modalities and over 30 cancer types.
 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.
 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.