syntax fix
Browse files- 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,
|