Initial Seg Data Server Net platform

scripts/sync_weights.py

@@ -0,0 +1,33 @@
+#!/usr/bin/env python3
+from __future__ import annotations
+
+import argparse
+import sys
+from pathlib import Path
+
+ROOT = Path(__file__).resolve().parents[1]
+BACKEND = ROOT / "backend"
+sys.path.insert(0, str(BACKEND))
+
+from app.modules.weights.service import sync_weights  # noqa: E402
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser(description="Sync Seg weight files into Seg_Data_Server_Net/weights.")
+    parser.add_argument("--mode", choices=["copy", "reflink", "hardlink"], default="copy")
+    parser.add_argument("--hash", action="store_true", help="calculate sha256 for every copied weight")
+    parser.add_argument("--no-skip-existing", action="store_true", help="recopy files even when size matches")
+    args = parser.parse_args()
+    manifest = sync_weights(
+        mode=args.mode,
+        hash_files=args.hash,
+        skip_existing=not args.no_skip_existing,
+    )
+    total_gb = manifest["total_bytes"] / 1024 / 1024 / 1024
+    print(f"synced {manifest['count']} weights, total {total_gb:.2f} GB")
+    print("manifest:", ROOT / "weights" / "manifest.json")
+
+
+if __name__ == "__main__":
+    main()
+