Update all files for SegEarth-OV

OV-2/pipeline.py

@@ -0,0 +1,38 @@
+"""
+SegEarth OV-2 pipeline. Self-contained; uses config.json and weights/featup/ in this folder.
+"""
+import json
+from pathlib import Path
+
+import sys
+_parent = Path(__file__).resolve().parent.parent
+if str(_parent) not in sys.path:
+    sys.path.insert(0, str(_parent))
+from pipeline import SegEarthPipelineCLIP
+
+
+def load(config_path: Path = None, model_id: str = None, **kwargs):
+    """Load OV-2 pipeline with config from this folder."""
+    repo_dir = Path(__file__).parent
+    cfg_path = config_path or repo_dir / "config.json"
+    with open(cfg_path) as f:
+        cfg = json.load(f)
+    local_backbone = cfg.get("local_backbone")
+    if local_backbone:
+        local_path = repo_dir / local_backbone
+        if local_path.exists():
+            kwargs.setdefault("model_id", str(local_path))
+    if "model_id" not in kwargs:
+        kwargs.setdefault("model_id", model_id or cfg["model_id"])
+    kwargs.setdefault("featup_model", cfg.get("featup") or "jbu_one")
+    kwargs.setdefault("cls_token_lambda", cfg.get("cls_token_lambda", -0.3))
+    kwargs.setdefault("logit_scale", cfg.get("logit_scale", 50.0))
+    featup_name = (cfg.get("featup_weights") or "xclip_jbu_one_million_aid.ckpt").split("/")[-1]
+    local_featup = repo_dir / "weights" / "featup" / featup_name
+    if local_featup.exists():
+        kwargs.setdefault("featup_weights_path", local_featup)
+    kwargs.setdefault("class_names_path", repo_dir / "configs" / "cls_openearthmap_sar.txt")
+    return SegEarthPipelineCLIP(**kwargs)
+
+
+SegEarthPipeline = load