jhj0517 commited on
Commit
c5a4e30
1 Parent(s): 0f36b51

Chore: Update variable name

Browse files
Files changed (1) hide show
  1. modules/sam_inference.py +2 -2
modules/sam_inference.py CHANGED
@@ -21,7 +21,7 @@ from modules.mask_utils import (
21
  )
22
  from modules.logger_util import get_logger
23
 
24
- CONFIGS = {
25
  "sam2_hiera_tiny": os.path.join(SAM2_CONFIGS_DIR, "sam2_hiera_t.yaml"),
26
  "sam2_hiera_small": os.path.join(SAM2_CONFIGS_DIR, "sam2_hiera_s.yaml"),
27
  "sam2_hiera_base_plus": os.path.join(SAM2_CONFIGS_DIR, "sam2_hiera_b+.yaml"),
@@ -48,7 +48,7 @@ class SamInference:
48
 
49
  def load_model(self,
50
  load_video_predictor: bool = False):
51
- config = CONFIGS[self.model_type]
52
  filename, url = AVAILABLE_MODELS[self.model_type]
53
  model_path = os.path.join(self.model_dir, filename)
54
 
 
21
  )
22
  from modules.logger_util import get_logger
23
 
24
+ MODEL_CONFIGS = {
25
  "sam2_hiera_tiny": os.path.join(SAM2_CONFIGS_DIR, "sam2_hiera_t.yaml"),
26
  "sam2_hiera_small": os.path.join(SAM2_CONFIGS_DIR, "sam2_hiera_s.yaml"),
27
  "sam2_hiera_base_plus": os.path.join(SAM2_CONFIGS_DIR, "sam2_hiera_b+.yaml"),
 
48
 
49
  def load_model(self,
50
  load_video_predictor: bool = False):
51
+ config = MODEL_CONFIGS[self.model_type]
52
  filename, url = AVAILABLE_MODELS[self.model_type]
53
  model_path = os.path.join(self.model_dir, filename)
54