opensportslib 0.1.2.dev8__tar.gz → 0.1.2.dev10__tar.gz

{opensportslib-0.1.2.dev8/opensportslib.egg-info → opensportslib-0.1.2.dev10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opensportslib
-Version: 0.1.2.dev8
+Version: 0.1.2.dev10
 Summary: OpenSportsLib is the professional library, designed for advanced video understanding in sports. It provides state-of-the-art tools for action recognition, spotting, retrieval, and captioning, making it ideal for researchers, analysts, and developers working with sports video data.
 Author: Jeet Vora
 Requires-Python: >=3.12

{opensportslib-0.1.2.dev8 → opensportslib-0.1.2.dev10}/opensportslib/datasets/classification_dataset.py

@@ -73,7 +73,7 @@ class ClassificationDataset(Dataset):
         self.config = config
         self.split = split
         self.exclude_labels = ["Unknown", "Dont know"]
-        self.data_dir = config.DATA.data_dir
+        self.video_path = getattr(config.DATA, split).video_path #config.DATA.data_dir
         self.processor = None
 
         # view_type is optional; only MVFoul uses it as of now
@@ -197,7 +197,7 @@ class VideoDataset(ClassificationDataset):
     """
 
     def __init__(self, config, annotations_path, processor, split="train"):
-        super().__init__(config, annotations_path, split)
+        super().__init__(config, annotations_path, processor, split=split)
 
         self.processor = processor
         self.view_type = getattr(config.DATA, "view_type", "single")
@@ -227,15 +227,13 @@ class VideoDataset(ClassificationDataset):
         """read a video file, temporally sub-sample, and apply transforms.
 
         Args:
-            path: realtive path (under data_dir) to the video file.
+            path: realtive path (under video_path) to the video file.
 
         Returns:
             numpy.ndarray of shape (T, H, W, C).
         """
-        full_path = os.path.join(self.config.DATA.data_dir, path)
-
-        if full_path.endswith(".npy"):
-            frames = np.load(full_path).astype(np.float32) / 255.0
+        if path.endswith(".npy"):
+            frames = np.load(path).astype(np.float32) / 255.0
             if self.transform is not None:
                 frames = self.transform(frames)
             mean = np.array([0.485, 0.456, 0.406], dtype=np.float32)
@@ -243,7 +241,7 @@ class VideoDataset(ClassificationDataset):
             frames = (frames - mean) / std
             return frames
 
-        v = read_video(os.path.join(self.config.DATA.data_dir, path))
+        v = read_video(os.path.join(self.video_path, path))
 
         v = process_frames(
             v,
@@ -473,7 +471,7 @@ class TrackingDataset(ClassificationDataset):
         """read a single parquet tracking clip.
         
         Args:
-            path: Relative path (under ``data_dir``) to the parquet
+            path: Relative path (under ``video_path``) to the parquet
                 file.
 
         Returns:
@@ -481,7 +479,7 @@ class TrackingDataset(ClassificationDataset):
         """
         import pandas as pd
 
-        full_path = os.path.join(self.data_dir, path)
+        full_path = os.path.join(self.video_path, path)
         return pd.read_parquet(full_path)
     
     def __getitem__(self, idx):

{opensportslib-0.1.2.dev8 → opensportslib-0.1.2.dev10/opensportslib.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opensportslib
-Version: 0.1.2.dev8
+Version: 0.1.2.dev10
 Summary: OpenSportsLib is the professional library, designed for advanced video understanding in sports. It provides state-of-the-art tools for action recognition, spotting, retrieval, and captioning, making it ideal for researchers, analysts, and developers working with sports video data.
 Author: Jeet Vora
 Requires-Python: >=3.12

{opensportslib-0.1.2.dev8 → opensportslib-0.1.2.dev10}/opensportslib.egg-info/SOURCES.txt

@@ -87,6 +87,7 @@ opensportslib/tools/hf_transfer.py
 opensportslib/tools/osl_json_to_parquet.py
 opensportslib/tools/parquet_to_osl_json.py
 tests/conftest.py
+tests/test_classification_dataset_paths.py
 tests/test_config_utils_smoke.py
 tests/test_conversion_tools.py
 tests/test_hf_transfer_tools.py
@@ -94,6 +95,8 @@ tests/test_package_smoke.py
 tests/test_public_apis_smoke.py
 tests/test_subset_train_infer_integration.py
 tests/test_task_model_api_contract.py
+tools/convert/build_soccernet_gar.py
+tools/convert/build_soccernet_gar_action_spotting.py
 tools/convert/osl_json_to_parquet_webdataset.py
 tools/convert/parquet_webdataset_to_osl_json.py
 tools/download/download_hf_repo.py

{opensportslib-0.1.2.dev8 → opensportslib-0.1.2.dev10}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "opensportslib"
-version = "0.1.2.dev8"
+version = "0.1.2.dev10"
 description = "OpenSportsLib is the professional library, designed for advanced video understanding in sports. It provides state-of-the-art tools for action recognition, spotting, retrieval, and captioning, making it ideal for researchers, analysts, and developers working with sports video data."
 readme = "README.md"
 requires-python = ">=3.12"

opensportslib-0.1.2.dev10/tests/test_classification_dataset_paths.py

@@ -0,0 +1,74 @@
+import json
+from pathlib import Path
+from types import SimpleNamespace
+
+import opensportslib.datasets.classification_dataset as classification_dataset
+
+
+def _write_classification_annotation(path: Path) -> str:
+    payload = {
+        "labels": {
+            "action": {"labels": ["PASS"]},
+        },
+        "data": [
+            {
+                "id": "sample_00000",
+                "inputs": [
+                    {
+                        "type": "video",
+                        "path": "clips/video_00000.mp4",
+                    }
+                ],
+                "labels": {
+                    "action": {"label": "PASS"},
+                },
+            }
+        ],
+    }
+    path.parent.mkdir(parents=True, exist_ok=True)
+    path.write_text(json.dumps(payload), encoding="utf-8")
+    return str(path)
+
+
+def _make_config(data_dir: Path, valid_video_root: Path) -> SimpleNamespace:
+    return SimpleNamespace(
+        DATA=SimpleNamespace(
+            data_dir=str(data_dir),
+            data_modality="video",
+            view_type="single",
+            num_frames=16,
+            input_fps=25,
+            target_fps=17,
+            start_frame=0,
+            end_frame=15,
+            train=SimpleNamespace(video_path=str(data_dir / "train_root")),
+            valid=SimpleNamespace(video_path=str(valid_video_root)),
+            test=SimpleNamespace(video_path=str(data_dir / "test_root")),
+        ),
+        MODEL=SimpleNamespace(type="custom"),
+    )
+
+
+def test_video_dataset_resolves_relative_paths_from_selected_split_root(
+    tmp_path,
+    monkeypatch,
+):
+    annotation_path = _write_classification_annotation(tmp_path / "annotations" / "valid.json")
+    data_dir = tmp_path / "dataset_root"
+    valid_video_root = tmp_path / "separate_valid_root"
+    config = _make_config(data_dir, valid_video_root)
+
+    monkeypatch.setattr(classification_dataset, "build_transform", lambda config, mode: None)
+
+    dataset = classification_dataset.VideoDataset(
+        config,
+        annotation_path,
+        processor=None,
+        split="valid",
+    )
+
+    resolved_path = Path(dataset.samples[0]["video_paths"][0])
+
+    assert dataset.split == "valid"
+    assert resolved_path.is_absolute()
+    assert resolved_path == valid_video_root / "clips" / "video_00000.mp4"