flaviagiammarino
/

medsam-vit-base

Mask Generation

Inference Endpoints

Model card Files Files and versions Community

flaviagiammarino commited on Jul 12, 2023

Commit

62750e5

•

1 Parent(s): a3833bb

Update scripts/pt_model.py

Files changed (1) hide show

scripts/pt_model.py +1 -1

scripts/pt_model.py CHANGED Viewed

@@ -4,7 +4,7 @@ from transformers.models.sam.convert_sam_original_to_hf_format import replace_ke
 from segment_anything import sam_model_registry  # pip install git+https://github.com/facebookresearch/segment-anything.git
 # load the MedSAM ViT-B model
-checkpoint = 'medsam_vit_b.pth'  # https://drive.google.com/file/d/1UAmWL88roYR7wKlnApw5Bcuzf2iQgk6_/view?usp=drive_link
 pt_model = sam_model_registry['vit_b'](checkpoint)
 pt_state_dict = pt_model.state_dict()

 from segment_anything import sam_model_registry  # pip install git+https://github.com/facebookresearch/segment-anything.git
 # load the MedSAM ViT-B model
+checkpoint = 'medsam_vit_b.pth'  # https://drive.google.com/drive/folders/1ETWmi4AiniJeWOt6HAsYgTjYv_fkgzoN?usp=drive_link
 pt_model = sam_model_registry['vit_b'](checkpoint)
 pt_state_dict = pt_model.state_dict()