From e94c8a0a5c99f6c8f5046f3838f923e2d52bcb8e Mon Sep 17 00:00:00 2001 From: brimoor Date: Fri, 20 Dec 2024 15:07:23 -0500 Subject: [PATCH] include recall_sweep in serialization --- fiftyone/utils/eval/coco.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/fiftyone/utils/eval/coco.py b/fiftyone/utils/eval/coco.py index bb3a6c6553..3de1702051 100644 --- 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, )