include recall_sweep in serialization

voxel51 · Dec 20, 2024 · e94c8a0 · e94c8a0
1 parent 770c7d6
commit e94c8a0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/fiftyone/utils/eval/coco.py b/fiftyone/utils/eval/coco.py
@@ -293,6 +293,7 @@ def __init__(
         self.precision = np.asarray(precision)
         self.recall = np.asarray(recall)
         self.iou_threshs = np.asarray(iou_threshs)
+        self.recall_sweep = recall_sweep
         self.thresholds = (
             np.asarray(thresholds) if thresholds is not None else None
         )
@@ -449,6 +450,7 @@ def _from_dict(cls, d, samples, config, eval_key, **kwargs):
         precision = d["precision"]
         recall = d["recall"]
         iou_threshs = d["iou_threshs"]
+        recall_sweep = d.get("recall_sweep", None)
         thresholds = d.get("thresholds", None)
         return super()._from_dict(
             d,
@@ -458,6 +460,7 @@ def _from_dict(cls, d, samples, config, eval_key, **kwargs):
             precision=precision,
             recall=recall,
             iou_threshs=iou_threshs,
+            recall_sweep=recall_sweep,
             thresholds=thresholds,
             **kwargs,
         )