Updating for readabilitiy

voxel51 · Dec 30, 2024 · 7c56ce7 · 7c56ce7
1 parent a167314
commit 7c56ce7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/fiftyone/utils/eval/segmentation.py b/fiftyone/utils/eval/segmentation.py
@@ -390,11 +390,13 @@ def evaluate_samples(
                     values,
                     bandwidth=bandwidth,
                 )
-                sample_conf_mat += image_conf_mat
+
                 non_zero_indexes = np.nonzero(image_conf_mat)
                 for index in zip(*non_zero_indexes):
                     matches.append((classes[index[0]], classes[index[1]], image_conf_mat[index[0], index[1]], gt_seg.id, pred_seg.id))
 
+                sample_conf_mat += image_conf_mat
+
                 if processing_frames and save:
                     facc, fpre, frec = _compute_accuracy_precision_recall(
                         image_conf_mat, values, average