voxel51 / fiftyone-brain

Open source AI/ML capabilities for the FiftyOne ecosystem
https://fiftyone.ai/brain.html
Apache License 2.0
128 stars 3 forks source link

Fix bug in mistakenness for unmatched detections #79

Closed ehofesmann closed 3 years ago

ehofesmann commented 3 years ago

Detection evaluation was updated so that eval_key + "_iou" no longer exists on false negatives which was causing a bug when computing detection mistakenness. This PR resolves that issue.

Previously this would fail:

import fiftyone.brain as fob
import fiftyone.zoo as foz

dataset = foz.load_zoo_dataset("quickstart")
fob.compute_mistakenness(dataset, "predictions", label_field="ground_truth")