Initial commit

2026-09-11 08:28:02 +03:00
commit 1f9b2f20bd
64 changed files with 12593 additions and 0 deletions
@@ -0,0 +1,55 @@
+from __future__ import annotations
+
+import json
+import sys
+from pathlib import Path
+
+
+PROJECT_ROOT = Path(__file__).resolve().parents[1]
+SRC_ROOT = PROJECT_ROOT / "src"
+VENDOR_ROOT = PROJECT_ROOT / "vendor"
+VENV_SITE_PACKAGES = sorted((PROJECT_ROOT / ".venv" / "lib").glob("python*/site-packages"))
+
+for candidate in [*VENV_SITE_PACKAGES, SRC_ROOT, VENDOR_ROOT]:
+    candidate_str = str(candidate)
+    if candidate.exists() and candidate_str not in sys.path:
+        sys.path.append(candidate_str)
+
+from dataset.build_patient_summary import build_patient_summary
+from dataset.build_subsets import build_subsets
+from dataset.build_tasks import build_tasks
+from dataset.make_group_splits import make_group_splits
+
+
+def main() -> None:
+    subset_result = build_subsets(
+        object_index_path=PROJECT_ROOT / "outputs" / "dataset_index" / "object_index.parquet",
+        image_index_path=PROJECT_ROOT / "outputs" / "dataset_index" / "image_index.parquet",
+    )
+    patient_summary_path = build_patient_summary(
+        patient_inventory_path=PROJECT_ROOT / "outputs" / "dataset_index" / "patient_inventory.csv",
+        image_index_path=PROJECT_ROOT / "outputs" / "dataset_index" / "image_index.parquet",
+        object_index_path=PROJECT_ROOT / "outputs" / "dataset_index" / "object_index.parquet",
+    )
+    split_result = make_group_splits(
+        patient_summary_path=patient_summary_path,
+        object_index_path=PROJECT_ROOT / "outputs" / "dataset_index" / "object_index.parquet",
+        random_state=42,
+    )
+    task_result = build_tasks(PROJECT_ROOT / "outputs" / "splits" / "split_v1.json")
+
+    print(
+        json.dumps(
+            {
+                "a1": subset_result,
+                "a2": {"patient_summary_csv": str(patient_summary_path), **split_result},
+                "a4": task_result,
+            },
+            ensure_ascii=False,
+            indent=2,
+        )
+    )
+
+
+if __name__ == "__main__":
+    main()