This commit is contained in:
Noah
2022-03-24 15:44:57 +08:00
+2 -2
View File
@@ -211,8 +211,8 @@ def evaluate_HID(data, dataset, metric='euc'):
with open(save_path, "w") as f: with open(save_path, "w") as f:
f.write("videoID,label\n") f.write("videoID,label\n")
for i in range(len(idx)): for i in range(len(idx)):
f.write("{},{},\n".format(probe_y[i], gallery_y[idx[i, 0]])) f.write("{},{}\n".format(probe_y[i], gallery_y[idx[i, 0]]))
print("GREW result saved to {}/{}".format(os.getcwd(), save_path)) print("HID result saved to {}/{}".format(os.getcwd(), save_path))
return return