From d61fb69bb5c523f075d854bb45d287c8d3ecc7b7 Mon Sep 17 00:00:00 2001 From: jp-x-g Date: Tue, 11 Jul 2023 11:14:03 -0700 Subject: [PATCH] Update README.md with checkpoint sizes Large sizes are linked without file sizes, which makes it difficult to select one. --- README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 4f5efb986..7f0a32e8e 100644 --- a/README.md +++ b/README.md @@ -97,9 +97,9 @@ sam = sam_model_registry[""](checkpoint="") Click the links below to download the checkpoint for the corresponding model type. -- **`default` or `vit_h`: [ViT-H SAM model.](https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth)** -- `vit_l`: [ViT-L SAM model.](https://dl.fbaipublicfiles.com/segment_anything/sam_vit_l_0b3195.pth) -- `vit_b`: [ViT-B SAM model.](https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth) + **`default` or `vit_h`: [ViT-H SAM model.](https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth)** (2.4 GB) +* `vit_l`: [ViT-L SAM model.](https://dl.fbaipublicfiles.com/segment_anything/sam_vit_l_0b3195.pth) (1.2 GB) +* `vit_b`: [ViT-B SAM model.](https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth) (358 MB) ## Dataset