Improve oks sigma comment

humanprotocol · Feb 9, 2024 · f452836 · f452836
1 parent 132dece
commit f452836
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/packages/examples/cvat/recording-oracle/src/validation/dataset_comparison.py b/packages/examples/cvat/recording-oracle/src/validation/dataset_comparison.py
@@ -142,9 +142,11 @@ def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
         self._skeleton_info: Dict[int, list[str]] = {}
-        self.oks_sigma = 0.1  # TODO: try to estimate from GT
         self.categories: Optional[dm.CategoriesInfo] = None
 
+        # TODO: find better strategy for sigma estimation
+        self.oks_sigma = 0.1  # average value for COCO points
+
     def compare(self, gt_dataset: dm.Dataset, ds_dataset: dm.Dataset) -> float:
         self.categories = gt_dataset.categories()