opensportslib 0.1.3.dev4__tar.gz → 0.2.0.dev1__tar.gz

{opensportslib-0.1.3.dev4/opensportslib.egg-info → opensportslib-0.2.0.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opensportslib
-Version: 0.1.3.dev4
+Version: 0.2.0.dev1
 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
@@ -257,7 +257,7 @@ print(metrics)
 from opensportslib.apis import LocalizationModel
 
 my_model = LocalizationModel(
-    config="/path/to/localization.yaml",
+    config="/path/to/localization_video_dali.yaml",
     weights=None,  # optional: path or Hugging Face model ID
 )
 

{opensportslib-0.1.3.dev4 → opensportslib-0.2.0.dev1}/README.md

@@ -222,7 +222,7 @@ print(metrics)
 from opensportslib.apis import LocalizationModel
 
 my_model = LocalizationModel(
-    config="/path/to/localization.yaml",
+    config="/path/to/localization_video_dali.yaml",
     weights=None,  # optional: path or Hugging Face model ID
 )
 

{opensportslib-0.1.3.dev4 → opensportslib-0.2.0.dev1}/examples/quickstart/basic_localization.py

@@ -8,7 +8,7 @@ def main():
     """
 
     my_model = LocalizationModel(
-        config="examples/configs/localization.yaml",
+        config="examples/configs/localization_video_dali.yaml",
         weights=None,  # optional: path or Hugging Face model ID
     )
 

{opensportslib-0.1.3.dev4 → opensportslib-0.2.0.dev1}/opensportslib/configs/classification/default.yaml

@@ -2,23 +2,13 @@ TASK: classification
 VERSION: 2
 
 SYSTEM:
-  paths:
-    log_dir: ./logs
-    save_dir: ./checkpoints
-  device: cuda
   gpu:
-    count: 1
-    id: 0
-  reproducibility:
-    use_seed: true
-    seed: 42
+    count: 4
 
 DATA:
   common:
     dataset_name: dataset_name
     data_root: /path/to/data
-    runtime:
-      loader_backend: opencv
     splits:
       train:
         annotation_path: /path/to/train.json
@@ -61,17 +51,6 @@ DATA:
         color_mode: rgb
 
 MODEL:
-  runtime:
-    dtype: fp32
-    device: auto
-    compile: false
-    freeze: false
-  load:
-    checkpoint_path: null
-    pretrained: false
-    strict: true
-    map_location: null
-    format: auto
   components:
     video_encoder:
       kind: encoder
@@ -81,6 +60,14 @@ MODEL:
         name: null
       params: {}
       overrides: {}
+    video_adapter:
+      kind: adapter
+      source:
+        provider: opensportslib
+        registry: neck
+        name: null
+      params: {}
+      overrides: {}
     task_head:
       kind: head
       source:
@@ -91,6 +78,8 @@ MODEL:
       overrides: {}
   topology:
     - from: video_encoder
+      to: video_adapter
+    - from: video_adapter
       to: task_head
 
 IO:

opensportslib-0.1.3.dev4/opensportslib/configs/classification/frames_npy/sngar-frames.yaml → opensportslib-0.2.0.dev1/opensportslib/configs/classification/sngar_frames.yaml

@@ -1,24 +1,11 @@
-TASK: classification
-VERSION: 2
-
 SYSTEM:
   paths:
-    log_dir: ./logs
     save_dir: ./checkpoints_video
-  device: cuda
-  gpu:
-    count: 4
-    id: 0
-  reproducibility:
-    use_seed: true
-    seed: 42
 
 DATA:
   common:
     dataset_name: sngar
     data_root: /home/vorajv/dataset/sngar-frames
-    runtime:
-      loader_backend: opencv
     splits:
       train:
         annotation_path: ${DATA.common.data_root}/train.json
@@ -69,20 +56,8 @@ DATA:
           training_matches: 45
 
 MODEL:
-  runtime:
-    dtype: fp32
-    device: auto
-    compile: false
-    freeze: false
-  load:
-    checkpoint_path: null
-    pretrained: false
-    strict: true
-    map_location: null
-    format: auto
   components:
     video_encoder:
-      kind: encoder
       source:
         provider: opensportslib
         registry: backbone
@@ -97,7 +72,6 @@ MODEL:
       overrides:
         unfreeze_last_n_layers: 0
     video_adapter:
-      kind: adapter
       source:
         provider: opensportslib
         registry: neck
@@ -108,9 +82,7 @@ MODEL:
         dropout: 0.1
         lstm_dropout: 0.3
         num_attention_heads: 8
-      overrides: {}
     task_head:
-      kind: head
       source:
         provider: opensportslib
         registry: head
@@ -119,26 +91,9 @@ MODEL:
         hidden_dim: 64
         num_classes: 10
         dropout: 0.1
-      overrides: {}
-  topology:
-    - from: video_encoder
-      to: video_adapter
-    - from: video_adapter
-      to: task_head
-
-IO:
-  inputs:
-    pixel_values: video_encoder
-    label: task_head
-  outputs:
-    logits: task_head
 
 TRAIN:
-  trainer:
-    type: classification
   epochs: 100
-  criterion:
-    type: CrossEntropyLoss
   optimizer:
     type: AdamW
     lr: 0.00005
@@ -153,19 +108,14 @@ TRAIN:
     factor: 0.1
     min_lr: 1e-8
   execution:
-    enabled: true
     use_amp: true
     mixup_alpha: 0.2
-    log_interval: 10
     detailed_results: true
   sampling:
     use_weighted_sampler: true
-    use_weighted_loss: false
     samples_per_class: 4000
   selection:
     monitor: loss
-    mode: min
     patience: 5
   checkpoint:
     save_every: 5
-    save_best: true

opensportslib-0.1.3.dev4/opensportslib/configs/classification/tracking/sngar-tracking.yaml → opensportslib-0.2.0.dev1/opensportslib/configs/classification/sngar_tracking.yaml

@@ -1,24 +1,11 @@
-TASK: classification
-VERSION: 2
-
 SYSTEM:
   paths:
-    log_dir: ./logs
     save_dir: ./checkpoints_tracking
-  device: cuda
-  gpu:
-    count: 4
-    id: 0
-  reproducibility:
-    use_seed: true
-    seed: 42
 
 DATA:
   common:
     dataset_name: sngar
     data_root: /home/vorajv/dataset/sngar-tracking
-    runtime:
-      loader_backend: opencv
     splits:
       train:
         type: annotations_train.json
@@ -48,7 +35,7 @@ DATA:
           num_workers: 8
           pin_memory: true
   inputs:
-    tracking:
+    video:
       modality: tracking_parquet
       representation: raw
       source:
@@ -76,20 +63,8 @@ DATA:
           training_matches: 45
 
 MODEL:
-  runtime:
-    dtype: fp32
-    device: auto
-    compile: false
-    freeze: false
-  load:
-    checkpoint_path: null
-    pretrained: false
-    strict: true
-    map_location: null
-    format: auto
   components:
-    tracking_encoder:
-      kind: encoder
+    video_encoder:
       source:
         provider: opensportslib
         registry: backbone
@@ -102,9 +77,7 @@ MODEL:
         edge_type: positional
         k: 8
         radius: 15.0
-      overrides: {}
-    tracking_adapter:
-      kind: adapter
+    video_adapter:
       source:
         provider: opensportslib
         registry: neck
@@ -114,9 +87,7 @@ MODEL:
         hidden_dim: 64
         dropout: 0.1
         use_position_encoding: true
-      overrides: {}
     task_head:
-      kind: head
       source:
         provider: opensportslib
         registry: head
@@ -125,26 +96,9 @@ MODEL:
         hidden_dim: 64
         dropout: 0.1
         num_classes: 10
-      overrides: {}
-  topology:
-    - from: tracking_encoder
-      to: tracking_adapter
-    - from: tracking_adapter
-      to: task_head
-
-IO:
-  inputs:
-    tracking: tracking_encoder
-    label: task_head
-  outputs:
-    logits: task_head
 
 TRAIN:
-  trainer:
-    type: classification
   epochs: 100
-  criterion:
-    type: CrossEntropyLoss
   optimizer:
     type: Adam
     lr: 0.001
@@ -155,16 +109,12 @@ TRAIN:
     factor: 0.1
     min_lr: 1e-8
   execution:
-    enabled: true
     detailed_results: true
   sampling:
     use_weighted_sampler: true
-    use_weighted_loss: false
     samples_per_class: 4000
   selection:
     monitor: loss
-    mode: min
     patience: 10
   checkpoint:
     save_every: 20
-    save_best: true

opensportslib-0.1.3.dev4/opensportslib/configs/classification/video/classification.yaml → opensportslib-0.2.0.dev1/opensportslib/configs/classification/video.yaml

@@ -1,24 +1,13 @@
-TASK: classification
-VERSION: 2
-
 SYSTEM:
   paths:
-    log_dir: ./logs
-    save_dir: ./checkpoints
-  device: cuda
-  gpu:
-    count: 4
-    id: 0
+    save_dir: ./checkpoints_video
   reproducibility:
     use_seed: false
-    seed: 42
 
 DATA:
   common:
     dataset_name: mvfouls
     data_root: /home/vorajv/dataset/OSL-XFoul
-    runtime:
-      loader_backend: opencv
     splits:
       train:
         type: annotations_train.json
@@ -58,20 +47,11 @@ DATA:
           prefetch_factor: 4
   inputs:
     video:
-      modality: video
-      representation: raw
-      source:
-        format: mp4
       sampling:
-        num_frames: 16
         input_fps: 25
         target_fps: 17
         start_frame: 63
         end_frame: 87
-      transform:
-        resize:
-          height: 224
-          width: 224
       augmentations:
         random_affine: true
         translate: [0.1, 0.1]
@@ -87,25 +67,12 @@ DATA:
         flip_prob: 0.5
         random_crop: false
       params:
-        color_mode: rgb
         view_type: multi
         num_classes: 8
 
 MODEL:
-  runtime:
-    dtype: fp32
-    device: auto
-    compile: false
-    freeze: false
-  load:
-    checkpoint_path: null
-    pretrained: false
-    strict: true
-    map_location: null
-    format: auto
   components:
     video_encoder:
-      kind: encoder
       source:
         provider: torchvision
         name: mvit_v2_s
@@ -115,42 +82,21 @@ MODEL:
         unfreeze_head: true
         unfreeze_last_n_layers: 3
     video_adapter:
-      kind: adapter
       source:
         provider: opensportslib
         registry: neck
         name: MV_Aggregate
       params:
         agr_type: max
-      overrides: {}
     task_head:
-      kind: head
       source:
         provider: opensportslib
         registry: head
         name: MV_LinearLayer
       params:
         num_classes: 8
-      overrides: {}
-  topology:
-    - from: video_encoder
-      to: video_adapter
-    - from: video_adapter
-      to: task_head
-
-IO:
-  inputs:
-    video: video_encoder
-    label: task_head
-  outputs:
-    logits: task_head
 
 TRAIN:
-  trainer:
-    type: classification
-  epochs: 20
-  criterion:
-    type: CrossEntropyLoss
   optimizer:
     type: AdamW
     lr: 0.0001
@@ -164,15 +110,10 @@ TRAIN:
     type: StepLR
     step_size: 3
     gamma: 0.1
-  execution:
-    enabled: true
-    log_interval: 10
   sampling:
-    use_weighted_sampler: false
     use_weighted_loss: true
   selection:
     monitor: balanced_accuracy
     mode: max
   checkpoint:
     save_every: 2
-    save_best: true

opensportslib-0.2.0.dev1/opensportslib/configs/default.yaml

@@ -0,0 +1,30 @@
+SYSTEM:
+  paths:
+    log_dir: ./logs
+    save_dir: ./checkpoints
+  device: cuda
+  gpu:
+    id: 0
+  reproducibility:
+    use_seed: true
+    seed: 42
+
+DATA:
+  common:
+    runtime:
+      loader_backend: opencv
+
+MODEL:
+  runtime:
+    dtype: fp32
+    device: auto
+    compile: false
+    freeze: false
+  load:
+    checkpoint_path: null
+    pretrained: false
+    strict: true
+    map_location: null
+    format: auto
+
+IO: {}

opensportslib-0.1.3.dev4/opensportslib/configs/localization/video_features/localization-calf-resnetpca512.yaml → opensportslib-0.2.0.dev1/opensportslib/configs/localization/calf_resnetpca512.yaml

@@ -1,19 +1,3 @@
-TASK: localization
-VERSION: 2
-
-SYSTEM:
-  paths:
-    log_dir: ./logs
-    save_dir: ./checkpoints
-    work_dir: ${SYSTEM.paths.save_dir}
-  device: cuda
-  gpu:
-    count: 4
-    id: 0
-  reproducibility:
-    use_seed: true
-    seed: 42
-
 DATA:
   common:
     dataset_name: SoccerNet
@@ -36,8 +20,6 @@ DATA:
       - Yellow card
       - Red card
       - Yellow->red card
-    runtime:
-      loader_backend: opencv
     splits:
       train:
         type: FeatureClipChunksfromJson
@@ -72,7 +54,7 @@ DATA:
           num_workers: 1
           pin_memory: true
   inputs:
-    features:
+    video:
       modality: video
       representation: features
       source:
@@ -87,24 +69,12 @@ DATA:
       params: {}
 
 MODEL:
-  metadata: 
+  metadata:
     family: ContextAware
-    runner: 
+    runner:
       type: runner_JSON
-  runtime:
-    dtype: fp32
-    device: auto
-    compile: false
-    freeze: false
-  load:
-    checkpoint_path: null
-    pretrained: false
-    strict: true
-    map_location: null
-    format: auto
   components:
-    feature_encoder:
-      kind: encoder
+    video_encoder:
       source:
         provider: opensportslib
         registry: backbone
@@ -114,9 +84,7 @@ MODEL:
         feature_dim: 512
         output_dim: 512
         framerate: 2
-      overrides: {}
-    feature_adapter:
-      kind: adapter
+    video_adapter:
       source:
         provider: opensportslib
         registry: neck
@@ -129,9 +97,7 @@ MODEL:
         receptive_field: 40
         num_detections: 15
         framerate: 2
-      overrides: {}
     task_head:
-      kind: head
       source:
         provider: opensportslib
         registry: head
@@ -141,17 +107,8 @@ MODEL:
         num_detections: 15
         chunk_size: 120
         num_classes: 17
-      overrides: {}
-  topology:
-    - from: feature_encoder
-      to: feature_adapter
-    - from: feature_adapter
-      to: task_head
 
 IO:
-  inputs:
-    features: feature_encoder
-    label: task_head
   outputs:
     predictions: task_head
 
@@ -191,11 +148,4 @@ TRAIN:
     patience: 25
     verbose: true
   execution:
-    enabled: true
     evaluation_frequency: 200
-  sampling: {}
-  selection:
-    monitor: valid_loss
-    mode: min
-  checkpoint:
-    save_best: true

{opensportslib-0.1.3.dev4 → opensportslib-0.2.0.dev1}/opensportslib/configs/localization/default.yaml

@@ -3,24 +3,15 @@ VERSION: 2
 
 SYSTEM:
   paths:
-    log_dir: ./logs
-    save_dir: ./checkpoints
     work_dir: ${SYSTEM.paths.save_dir}
-  device: cuda
   gpu:
     count: 4
-    id: 0
-  reproducibility:
-    use_seed: true
-    seed: 42
 
 DATA:
   common:
     dataset_name: SoccerNet
     data_root: /path/to/data
     classes: []
-    runtime:
-      loader_backend: opencv
     splits:
       train:
         annotation_path: /path/to/train.json
@@ -74,17 +65,10 @@ DATA:
         dilate_len: 0
 
 MODEL:
-  runtime:
-    dtype: fp32
-    device: auto
-    compile: false
-    freeze: false
-  load:
-    checkpoint_path: null
-    pretrained: false
-    strict: true
-    map_location: null
-    format: auto
+  metadata:
+    family: E2E
+    runner:
+      type: runner_e2e
   components:
     video_encoder:
       kind: encoder
@@ -115,7 +99,7 @@ IO:
 
 TRAIN:
   trainer:
-    type: localization
+    type: trainer_e2e
   epochs: 10
   criterion:
     type: CrossEntropyLoss
@@ -124,8 +108,17 @@ TRAIN:
     lr: 0.01
   scheduler:
     type: ChainedSchedulerE2E
+    acc_grad_iter: 1
+    num_epochs: ${TRAIN.epochs}
+    warm_up_epochs: 3
   execution:
     enabled: true
+    multi_gpu: true
+    acc_grad_iter: 1
+    base_num_valid_epochs: 30
+    start_valid_epoch: 4
+    valid_map_every: 1
+    criterion_valid: map
   sampling: {}
   selection:
     monitor: valid_loss