opensportslib 0.3.1.dev16__tar.gz → 0.3.1.dev17__tar.gz

{opensportslib-0.3.1.dev16/opensportslib.egg-info → opensportslib-0.3.1.dev17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: opensportslib
-Version: 0.3.1.dev16
+Version: 0.3.1.dev17
 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.1.dev16 → opensportslib-0.3.1.dev17}/opensportslib/adaptation/spotta.py

@@ -59,6 +59,7 @@ class SpoTTAConfig:
     drift_scale: float = 10.0
     drift_threshold: float = 1.0
     drift_gamma: float = 0.2
+    augmentation: bool = True
 
     @classmethod
     def from_mapping(cls, value: Any) -> "SpoTTAConfig":
@@ -69,6 +70,7 @@ class SpoTTAConfig:
         memory = _mapping(root.get("memory"))
         optimizer = _mapping(root.get("optimizer"))
         teacher = _mapping(root.get("teacher"))
+        augmentation = _mapping(root.get("augmentation"))
 
         _require_supported_value(
             root,
@@ -76,6 +78,34 @@ class SpoTTAConfig:
             "adapt_then_predict",
             "adaptation",
         )
+        _require_supported_value(tether, "mode", "bayesian", "robust_bn.tether")
+        _require_supported_value(
+            gate,
+            "uncertainty",
+            "one_minus_max_probability",
+            "confidence_gate",
+        )
+        _require_supported_value(
+            gate,
+            "aggregation",
+            "min_over_predicted_action_frames",
+            "confidence_gate",
+        )
+        _require_supported_value(optimizer, "type", "Adam", "optimizer")
+        _require_supported_value(
+            optimizer,
+            "trainable_parameters",
+            "batch_norm_affine_only",
+            "optimizer",
+        )
+        _require_supported_value(teacher, "type", "ema", "teacher")
+        _require_supported_value(
+            teacher, "adaptive_from_bn_drift", True, "teacher"
+        )
+        _require_supported_value(
+            augmentation, "mode", "framewise_rotta_strong", "augmentation"
+        )
+
         config = cls(
             alpha=float(robust_bn.get("alpha", cls.alpha)),
             tether_cap=float(tether.get("cap", cls.tether_cap)),
@@ -103,6 +133,7 @@ class SpoTTAConfig:
                 teacher.get("drift_threshold", cls.drift_threshold)
             ),
             drift_gamma=float(teacher.get("drift_gamma", cls.drift_gamma)),
+            augmentation=bool(augmentation.get("enabled", cls.augmentation)),
         )
         config.validate()
         return config
@@ -546,10 +577,13 @@ class SpoTTA:
         with torch.no_grad():
             teacher_logits = _logits(self.teacher(clips))
 
-        batch, time, channels, height, width = clips.shape
-        augmented = self.augmentation(
-            clips.reshape(batch * time, channels, height, width)
-        ).reshape(batch, time, channels, height, width)
+        if self.config.augmentation:
+            batch, time, channels, height, width = clips.shape
+            augmented = self.augmentation(
+                clips.reshape(batch * time, channels, height, width)
+            ).reshape(batch, time, channels, height, width)
+        else:
+            augmented = clips
 
         self.student.train()
         student_logits = _logits(self.student(augmented))

{opensportslib-0.3.1.dev16 → opensportslib-0.3.1.dev17}/opensportslib/configs/localization/e2e_spotta.yaml

@@ -39,7 +39,6 @@ DATA:
         resize:
           height: 224
           width: 398
-          preserve_aspect_ratio: true
       params:
         crop_dim: 224
         dilate_len: 0
@@ -70,10 +69,13 @@ MODEL:
       robust_bn:
         alpha: 0.05
         tether:
+          mode: bayesian
           cap: 0.5
       confidence_gate:
         action_class_index: 1
         min_action_frames: 1
+        uncertainty: one_minus_max_probability
+        aggregation: min_over_predicted_action_frames
         threshold: 0.3
       memory:
         capacity: 8
@@ -81,14 +83,21 @@ MODEL:
         lambda_t: 1.0
         lambda_u: 1.0
       optimizer:
+        type: Adam
         learning_rate: 0.001
         beta: 0.9
+        trainable_parameters: batch_norm_affine_only
       teacher:
+        type: ema
         base_nu: 0.001
+        adaptive_from_bn_drift: true
         max_nu: 0.02
         drift_scale: 10.0
         drift_threshold: 1.0
         drift_gamma: 0.2
+      augmentation:
+        enabled: true
+        mode: framewise_rotta_strong
 
 TRAIN:
   execution:

{opensportslib-0.3.1.dev16 → opensportslib-0.3.1.dev17/opensportslib.egg-info}/PKG-INFO

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

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