opensportslib 0.0.1.dev10__tar.gz → 0.0.1.dev12__tar.gz

{opensportslib-0.0.1.dev10/opensportslib.egg-info → opensportslib-0.0.1.dev12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opensportslib
-Version: 0.0.1.dev10
+Version: 0.0.1.dev12
 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.0.1.dev10 → opensportslib-0.0.1.dev12}/opensportslib/apis/localization.py

@@ -206,7 +206,7 @@ class LocalizationAPI:
             # # Inference
             inferer = build_inferer(cfg=self.config.MODEL,
                                     model=self.model)
-            json_gz_file = inferer.infer(cfg=self.config, data=test_loader)
+            json_gz_file = inferer.infer(cfg=self.config, data=dataset_Test)
 
         #json_gz_file = self.config.DATA.test.results + ".recall.json.gz"
         json_gz_file = predictions if predictions else json_gz_file

opensportslib-0.0.1.dev12/opensportslib/config/localization-e2e-ocv.yaml

@@ -0,0 +1,141 @@
+TASK: localization
+
+dali: false
+
+DATA:
+  dataset_name: SoccerNet
+  data_dir: /home/vorajv/opensportslib/SoccerNet/annotations/
+  classes:
+    - PASS
+    - DRIVE
+    - HEADER
+    - HIGH PASS
+    - OUT
+    - CROSS
+    - THROW IN
+    - SHOT
+    - BALL PLAYER BLOCK
+    - PLAYER SUCCESSFUL TACKLE
+    - FREE KICK
+    - GOAL
+    
+  epoch_num_frames: 500000
+  mixup: true
+  modality: rgb
+  crop_dim: -1
+  dilate_len: 0        # Dilate ground truth labels
+  clip_len: 100
+  input_fps: 25
+  extract_fps: 2
+  imagenet_mean: [0.485, 0.456, 0.406]
+  imagenet_std: [0.229, 0.224, 0.225]
+  target_height: 224
+  target_width: 398
+
+  train:
+    type: VideoGameWithOpencv
+    classes: ${DATA.classes}
+    output_map: [data, label]
+    video_path: ${DATA.data_dir}/train/
+    path: ${DATA.train.video_path}/annotations-2024-224p-train.json
+    dataloader:
+      batch_size: 8
+      shuffle: true
+      num_workers: 4
+      pin_memory: true
+      prefetch_factor: 1
+
+  valid:
+    type: VideoGameWithOpencv
+    classes: ${DATA.classes}
+    output_map: [data, label]
+    video_path: ${DATA.data_dir}/valid/
+    path: ${DATA.valid.video_path}/annotations-2024-224p-valid.json
+    dataloader:
+      batch_size: 8
+      shuffle: false
+      num_workers: 4
+      pin_memory: true
+      prefetch_factor: 1
+
+  valid_data_frames:
+    type: VideoGameWithOpencvVideo
+    classes: ${DATA.classes}
+    output_map: [data, label]
+    video_path: ${DATA.valid.video_path}
+    path: ${DATA.valid.path}
+    overlap_len: 0
+    dataloader:
+      batch_size: 8
+      shuffle: false
+      num_workers: 4
+      pin_memory: true
+      prefetch_factor: 1
+
+  test:
+    type: VideoGameWithOpencvVideo
+    classes: ${DATA.classes}
+    output_map: [data, label]
+    video_path: ${DATA.data_dir}/test/
+    path: ${DATA.test.video_path}/annotations-2024-224p-test.json
+    results: results_spotting_test
+    nms_window: 2 
+    metric: tight
+    overlap_len: 50
+    dataloader:
+      batch_size: 8
+      shuffle: false
+      num_workers: 4
+      pin_memory: true
+
+  challenge:
+    type: VideoGameWithOpencvVideo
+    overlap_len: 50
+    output_map: [data, label]
+    path: ${DATA.data_dir}/challenge/annotations.json
+    dataloader:
+      batch_size: 4
+      shuffle: false
+
+MODEL:
+    type: E2E
+    runner:
+      type: runner_e2e
+    backbone:
+      type: rny008_gsm
+    head:
+      type: gru
+    multi_gpu: true
+    load_weights: null
+
+TRAIN:
+  type: trainer_e2e
+  num_epochs: 10
+  acc_grad_iter: 1
+  base_num_valid_epochs: 30
+  start_valid_epoch: 4
+  valid_map_every: 1
+  criterion_valid: map
+
+  criterion:
+    type: CrossEntropyLoss
+
+  optimizer:
+    type: AdamWithScaler
+    lr: 0.01
+
+  scheduler:
+    type: ChainedSchedulerE2E
+    acc_grad_iter: 1
+    num_epochs: ${TRAIN.num_epochs}
+    warm_up_epochs: 3
+
+SYSTEM:
+  log_dir: ./logs
+  save_dir: ./checkpoints
+  work_dir: ${SYSTEM.save_dir}
+  seed: 42
+  GPU: 4         # number of gpus to use
+  device: cuda   # auto | cuda | cpu
+  gpu_id: 0      # device id for single gpu training
+

{opensportslib-0.0.1.dev10 → opensportslib-0.0.1.dev12/opensportslib.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opensportslib
-Version: 0.0.1.dev10
+Version: 0.0.1.dev12
 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.0.1.dev10 → opensportslib-0.0.1.dev12}/opensportslib.egg-info/SOURCES.txt

@@ -16,6 +16,7 @@ opensportslib/apis/classification.py
 opensportslib/apis/localization.py
 opensportslib/config/classification.yaml
 opensportslib/config/classification_tracking.yaml
+opensportslib/config/localization-e2e-ocv.yaml
 opensportslib/config/localization.yaml
 opensportslib/config/sngar_frames.yaml
 opensportslib/config/graph_tracking_classification/avgpool.yaml

{opensportslib-0.0.1.dev10 → opensportslib-0.0.1.dev12}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "opensportslib"
-version = "0.0.1.dev10"
+version = "0.0.1.dev12"
 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"