include recall_sweep in serialization

voxel51 · Dec 20, 2024 · 2214753 · 2214753
1 parent 770c7d6
commit 2214753
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/fiftyone/utils/eval/coco.py b/fiftyone/utils/eval/coco.py
@@ -449,6 +449,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 +459,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,
         )