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