diff --git a/tools/checkpoint_conversion/convert_sam_checkpoints.py b/tools/checkpoint_conversion/convert_sam_checkpoints.py index 08f4f4a504..69cd1482cb 100644 --- a/tools/checkpoint_conversion/convert_sam_checkpoints.py +++ b/tools/checkpoint_conversion/convert_sam_checkpoints.py @@ -1,3 +1,7 @@ +# Get the huge PyTorch model weights from the following location +# curl -sSL https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth -o sam_vit_h_4b8939.pth +# curl -sSL https://dl.fbaipublicfiles.com/segment_anything/sam_vit_l_0b3195.pth -o sam_vit_l_0b3195.pth +# curl -sSL https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth -o sam_vit_b_01ec64.pth import argparse import os