SkalskiP commited on
Commit
2c36ec6
1 Parent(s): 0f9d139

migrating SAM2 space from T4 to ZERO

Browse files
Files changed (1) hide show
  1. utils/models.py +1 -2
utils/models.py CHANGED
@@ -6,9 +6,8 @@ from sam2.build_sam import build_sam2
6
  from sam2.sam2_image_predictor import SAM2ImagePredictor
7
 
8
  BOX_PROMPT_MODE = "box prompt"
9
- POINT_PROMPT_MODE = "point prompt"
10
  MASK_GENERATION_MODE = "mask generation"
11
- MODE_NAMES = [BOX_PROMPT_MODE, POINT_PROMPT_MODE, MASK_GENERATION_MODE]
12
 
13
  CHECKPOINT_NAMES = ["tiny", "small", "base_plus", "large"]
14
  CHECKPOINTS = {
 
6
  from sam2.sam2_image_predictor import SAM2ImagePredictor
7
 
8
  BOX_PROMPT_MODE = "box prompt"
 
9
  MASK_GENERATION_MODE = "mask generation"
10
+ MODE_NAMES = [BOX_PROMPT_MODE, MASK_GENERATION_MODE]
11
 
12
  CHECKPOINT_NAMES = ["tiny", "small", "base_plus", "large"]
13
  CHECKPOINTS = {