MogensR commited on
Commit
4106ffb
·
1 Parent(s): 9bdec34

syntax fix

Browse files
Files changed (1) hide show
  1. models/__init__.py +16 -0
models/__init__.py CHANGED
@@ -404,6 +404,22 @@ def _try_build(cfg_path: str):
404
  sam = _try_build(cfg)
405
  except Exception:
406
  alt_cfg = _find_hiera_config_if_hieradet(cfg)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
407
 
408
  def run_sam2_mask(predictor: object,
409
  first_frame_bgr: np.ndarray,
 
404
  sam = _try_build(cfg)
405
  except Exception:
406
  alt_cfg = _find_hiera_config_if_hieradet(cfg)
407
+ if alt_cfg:
408
+ sam = _try_build(alt_cfg)
409
+ else:
410
+ raise
411
+
412
+ if sam is not None:
413
+ predictor = SAM2ImagePredictor(sam)
414
+ meta["sam2_init_ok"] = True
415
+ meta["sam2_device"] = device
416
+ return predictor, True, meta
417
+ else:
418
+ return None, False, meta
419
+
420
+ except Exception as e:
421
+ logger.error(f"SAM2 loading failed: {e}")
422
+ return None, False, meta
423
 
424
  def run_sam2_mask(predictor: object,
425
  first_frame_bgr: np.ndarray,