From 07c33780921c88445cf6e489e68a7bf3b43a220f Mon Sep 17 00:00:00 2001 From: niclaswue Date: Mon, 13 Jun 2022 16:01:42 +0200 Subject: [PATCH] Fix dataframe indexing single object samples --- objdetecteval/metrics/image_metrics.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/objdetecteval/metrics/image_metrics.py b/objdetecteval/metrics/image_metrics.py index 5c0b5f7..1b8742d 100644 --- a/objdetecteval/metrics/image_metrics.py +++ b/objdetecteval/metrics/image_metrics.py @@ -57,8 +57,8 @@ def match_preds_to_targets(predictions_df, labels_df): for image_name in unique_images: # get the predictions and labels for each image - preds = preds_df_indexed.loc[image_name] - labels = labels_df_indexed.loc[image_name] + preds = preds_df_indexed.loc[[image_name]] + labels = labels_df_indexed.loc[[image_name]] # create lists for all the bounding boxes, labels and scores # for the image, pascal boxes