opensportslib 0.3.0.dev5__tar.gz → 0.3.0.dev6__tar.gz

{opensportslib-0.3.0.dev5/opensportslib.egg-info → opensportslib-0.3.0.dev6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opensportslib
-Version: 0.3.0.dev5
+Version: 0.3.0.dev6
 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.3.0.dev5 → opensportslib-0.3.0.dev6}/opensportslib/datasets/localization_dataset.py

@@ -74,6 +74,15 @@ def _build_dali_filenames_and_labels(labels):
     return filenames, label_indices
 
 
+def _normalize_dali_devices(devices, sample_count):
+    devices = list(devices or [])
+    if not devices:
+        return [0]
+    if sample_count <= 0:
+        return devices[:1]
+    return devices[: max(1, min(len(devices), int(sample_count)))]
+
+
 def _dali_frame_num_to_local_frame(frame_num, stride):
     return frame_num // stride + 1
 
@@ -1538,13 +1547,16 @@ if DALI_AVAILABLE:
             else:
                 self.batch_size = batch_size
 
-            self.batch_size_per_pipe = distribute_elements(self.batch_size, len(devices))
+            self._filenames, self._video_indices = _build_dali_filenames_and_labels(
+                self._labels
+            )
+            self.devices = _normalize_dali_devices(devices, len(self._filenames))
+            self.batch_size_per_pipe = distribute_elements(self.batch_size, len(self.devices))
 
             self.batch_size = batch_size
             self.nb_videos = dataset_len * 2 if mixup else dataset_len
             self.mixup = mixup
             self.output_map = output_map
-            self.devices = devices
             self.is_eval = is_eval
             self.crop_dim = crop_dim
             self.dilate_len = dilate_len
@@ -1555,9 +1567,6 @@ if DALI_AVAILABLE:
             self.TARGET_WIDTH = TARGET_WIDTH
 
             self._stride = get_stride(input_fps, extract_fps)
-            self._filenames, self._video_indices = _build_dali_filenames_and_labels(
-                self._labels
-            )
 
             self.pipes = [
                 self.video_pipe(
@@ -1570,9 +1579,9 @@ if DALI_AVAILABLE:
                     num_threads=8,
                     device_id=i,
                     shard_id=index,
-                    num_shards=len(devices),
+                    num_shards=len(self.devices),
                 )
-                for index, i in enumerate(devices)
+                for index, i in enumerate(self.devices)
             ]
 
             for pipe in self.pipes:
@@ -1931,16 +1940,16 @@ if DALI_AVAILABLE:
             self._stride_dali = stride_dali
             self._flip = flip
             self._multi_crop = multi_crop
-            self.batch_size = batch_size // len(devices)
+            self._filenames, self._video_indices = _build_dali_filenames_and_labels(
+                self._labels
+            )
+            self.devices = _normalize_dali_devices(devices, len(self._filenames))
+            self.batch_size = batch_size // len(self.devices)
             self.global_batch_size = batch_size
-            self.devices = devices
             self.IMAGENET_MEAN = IMAGENET_MEAN
             self.IMAGENET_STD = IMAGENET_STD
             self.TARGET_HEIGHT = TARGET_HEIGHT
             self.TARGET_WIDTH = TARGET_WIDTH
-            self._filenames, self._video_indices = _build_dali_filenames_and_labels(
-                self._labels
-            )
             clip_count = 0
             for video in self._labels:
                 num_clips = _count_dali_video_samples(
@@ -1961,9 +1970,9 @@ if DALI_AVAILABLE:
                     num_threads=8,
                     device_id=i,
                     shard_id=index,
-                    num_shards=len(devices),
+                    num_shards=len(self.devices),
                 )
-                for index, i in enumerate(devices)
+                for index, i in enumerate(self.devices)
             ]
 
             for pipe in self.pipes:

{opensportslib-0.3.0.dev5 → opensportslib-0.3.0.dev6/opensportslib.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opensportslib
-Version: 0.3.0.dev5
+Version: 0.3.0.dev6
 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.3.0.dev5 → opensportslib-0.3.0.dev6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "opensportslib"
-version = "0.3.0.dev5"
+version = "0.3.0.dev6"
 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.3.0.dev5 → opensportslib-0.3.0.dev6}/tests/test_localization_dali_filenames.py

@@ -4,6 +4,7 @@ from opensportslib.datasets.localization_dataset import (
     _build_dali_filenames_and_labels,
     _count_dali_video_samples,
     _dali_frame_num_to_local_frame,
+    _normalize_dali_devices,
     _pad_dali_iterator_size,
     _resolve_dali_video_sample,
 )
@@ -56,4 +57,10 @@ def test_resolve_dali_video_sample_uses_relative_path_and_one_based_start():
     video_name, start = _resolve_dali_video_sample(labels, 0, 24, 12)
 
     assert video_name == "train/match one.mp4"
-    assert start == 3
+    assert start == 3
+
+
+def test_normalize_dali_devices_caps_shards_to_sample_count():
+    assert _normalize_dali_devices([0, 1, 2, 3], 1) == [0]
+    assert _normalize_dali_devices([0, 1, 2, 3], 2) == [0, 1]
+    assert _normalize_dali_devices([], 1) == [0]