search

facebookresearch / segment-anything-2

The repository provides code for running inference with the Meta Segment Anything Model 2 (SAM 2), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.
Apache License 2.0
10.71k stars 861 forks source link

MissingConfigException, what's the cfg name for sam2_hiera_base_plus.pt? #254

Closed yichuan1998 closed 3 days ago

yichuan1998 commented 3 weeks ago

It will be nice if there is a table which lists the cfg names for all .pt files image

Antuainy commented 3 weeks ago

admin_depext,+11+RAIMUNDO+-+NOVO.pdf

Pelirrojo commented 1 week ago

Hi,

The file is in the repo: sam2_hiera_b+.yaml

so: value sam2_configs/sam2_hiera_b+.yaml


# TINY
#sam2_checkpoint = "../checkpoints/sam2_hiera_tiny.pt"
#model_cfg = "sam2_hiera_t.yaml"

# BASE PLUS
#sam2_checkpoint = "../checkpoints/sam2_hiera_base_plus.pt"
#model_cfg = "sam2_hiera_b+.yaml"

# SMALL
sam2_checkpoint = "../checkpoints/sam2_hiera_small.pt"
model_cfg = "sam2_hiera_s.yaml"

# LARGE
#sam2_checkpoint = "../checkpoints/sam2_hiera_large.pt"
#model_cfg = "sam2_hiera_l.yaml"
yichuan1998 commented 1 week ago

Hi,

The file is in the repo: sam2_hiera_b+.yaml

so: value sam2_configs/sam2_hiera_b+.yaml

# TINY
#sam2_checkpoint = "../checkpoints/sam2_hiera_tiny.pt"
#model_cfg = "sam2_hiera_t.yaml"

# BASE PLUS
#sam2_checkpoint = "../checkpoints/sam2_hiera_base_plus.pt"
#model_cfg = "sam2_hiera_b+.yaml"

# SMALL
sam2_checkpoint = "../checkpoints/sam2_hiera_small.pt"
model_cfg = "sam2_hiera_s.yaml"

# LARGE
#sam2_checkpoint = "../checkpoints/sam2_hiera_large.pt"
#model_cfg = "sam2_hiera_l.yaml"

thanks!