From 76ea23774906e546fe9ecf1d02ae1ed277fe8e0a Mon Sep 17 00:00:00 2001 From: noahshen98 <77523610+noahshen98@users.noreply.github.com> Date: Wed, 15 Feb 2023 19:26:57 +0800 Subject: [PATCH] Update evaluator.py --- opengait/evaluation/evaluator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/opengait/evaluation/evaluator.py b/opengait/evaluation/evaluator.py index 411b00c..a25b83c 100644 --- a/opengait/evaluation/evaluator.py +++ b/opengait/evaluation/evaluator.py @@ -192,7 +192,7 @@ def GREW_submission(data, dataset, metric='euc'): def HID_submission(data, dataset, metric='euc'): msg_mgr = get_msg_mgr() msg_mgr.log_info("Evaluating HID") - feature, label, seq_type = data['embeddings'], data['labels'], data['types'] + feature, label, seq_type = data['embeddings'], data['labels'], data['views'] label = np.array(label) seq_type = np.array(seq_type) probe_mask = (label == "probe") @@ -264,4 +264,4 @@ def evaluate_Gait3D(data, conf, metric='euc'): # print_csv_format(dataset_name, results) msg_mgr.log_info(results) - return results \ No newline at end of file + return results