opensportslib 0.0.1.dev17__tar.gz → 0.0.1.dev18__tar.gz

{opensportslib-0.0.1.dev17/opensportslib.egg-info → opensportslib-0.0.1.dev18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opensportslib
-Version: 0.0.1.dev17
+Version: 0.0.1.dev18
 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.dev17 → opensportslib-0.0.1.dev18}/opensportslib/config/localization-json_netvlad++_resnetpca512.yaml

@@ -109,8 +109,8 @@ MODEL:
 
 TRAIN:
   type: trainer_pooling
-  max_epochs: 1000
-  evaluation_frequency: 1000
+  max_epochs: 200
+  evaluation_frequency: 200
   framerate: 2
   batch_size: 256
 

{opensportslib-0.0.1.dev17 → opensportslib-0.0.1.dev18}/opensportslib/core/trainer/localization_trainer.py

@@ -160,10 +160,17 @@ class Trainer_pl(Trainer):
     def __init__(self, cfg, work_dir):
         from opensportslib.core.utils.lightning import CustomProgressBar, MyCallback
         import pytorch_lightning as pl
+        from pytorch_lightning.loggers import WandbLogger
+        import wandb
+
+        wandb_logger = None
+        if wandb.run is not None:  # means init_wandb already ran
+            wandb_logger = WandbLogger(experiment=wandb.run)
 
         self.work_dir = work_dir
         call = MyCallback()
         self.trainer = pl.Trainer(
+            logger=wandb_logger,
             max_epochs=cfg.TRAIN.max_epochs,
             devices=cfg.SYSTEM.GPU,
             callbacks=[call, CustomProgressBar(refresh_rate=1)],
@@ -815,7 +822,6 @@ class Evaluator:
 
             # ---------------- GT ----------------
             if gt_is_v2:
-                print("Game: ", game)
                 video_path = game["inputs"][0]["path"]
                 labels = [{"label": e.get("label"),  
                            "gameTime": e.get("gameTime"),

{opensportslib-0.0.1.dev17 → opensportslib-0.0.1.dev18}/opensportslib/metrics/localization_metric.py

@@ -880,9 +880,9 @@ def label2vector(
 
         else:
             time = annotation["gameTime"]
-            print(time)
-            minutes = int(time[-5:-3])
-            seconds = int(time[-2::])
+            if time is not None:
+                minutes = int(time[-5:-3])
+                seconds = int(time[-2::])
             # annotation at millisecond precision
             if "position" in annotation:
                 frame = int(framerate * (int(annotation["position"]) / 1000))
@@ -931,7 +931,7 @@ def predictions2vector(
 
         event = annotation["label"]
 
-        if "frame" in annotation:
+        if "frame" in annotation and annotation["frame"] is not None:
             frame = int(annotation["frame"])
         else:
             time = int(annotation["position"])

{opensportslib-0.0.1.dev17 → opensportslib-0.0.1.dev18/opensportslib.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opensportslib
-Version: 0.0.1.dev17
+Version: 0.0.1.dev18
 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.dev17 → opensportslib-0.0.1.dev18}/pyproject.toml

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