diff --git a/lib/utils/evaluation.py b/lib/utils/evaluation.py index 9a5e4e6..335bcc8 100644 --- a/lib/utils/evaluation.py +++ b/lib/utils/evaluation.py @@ -211,8 +211,8 @@ def evaluate_HID(data, dataset, metric='euc'): with open(save_path, "w") as f: f.write("videoID,label\n") for i in range(len(idx)): - f.write("{},{},\n".format(probe_y[i], gallery_y[idx[i, 0]])) - print("GREW result saved to {}/{}".format(os.getcwd(), save_path)) + f.write("{},{}\n".format(probe_y[i], gallery_y[idx[i, 0]])) + print("HID result saved to {}/{}".format(os.getcwd(), save_path)) return