crosstyan/OpenGait
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update evaluator.py

Browse Source
This commit is contained in:
noahshen98
2023-02-15 19:26:57 +08:00
committed by GitHub
parent 311bc848fa
commit 76ea237749
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
opengait/evaluation/evaluator.py
+2 -2
View File
@@ -192,7 +192,7 @@ def GREW_submission(data, dataset, metric='euc'):
def HID_submission(data, dataset, metric='euc'): def HID_submission(data, dataset, metric='euc'):
msg_mgr = get_msg_mgr() msg_mgr = get_msg_mgr()
msg_mgr.log_info("Evaluating HID") 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) label = np.array(label)
seq_type = np.array(seq_type) seq_type = np.array(seq_type)
probe_mask = (label == "probe") probe_mask = (label == "probe")
@@ -264,4 +264,4 @@ def evaluate_Gait3D(data, conf, metric='euc'):
# print_csv_format(dataset_name, results) # print_csv_format(dataset_name, results)
msg_mgr.log_info(results) msg_mgr.log_info(results)
return results return results