Vincentqyw
fix: roma
c74a070
raw
history blame
3.1 kB
import os
import numpy as np
import subprocess
# def ratio(losses, thresholds=[1,2,3,4,5,6,7,8,9,10]):
def ratio(losses, thresholds=[5, 10]):
return ["{:.3f}".format(np.mean(losses < threshold)) for threshold in thresholds]
if __name__ == "__main__":
scene = "Indoor"
dir_base = "result_errors/Indoor/"
save_pt = "resultfinal_errors/Indoor/"
subprocess.check_output(["mkdir", "-p", save_pt])
with open(save_pt + "ratio_methods_" + scene + ".txt", "w") as f:
f.write("5deg 10deg" + "\n")
pair_list = os.listdir(dir_base)
enhancer = os.listdir(dir_base + "/pair9/")
for method in enhancer:
pose_error_list = sorted(os.listdir(dir_base + "/pair9/" + method))
for pose_error in pose_error_list:
error_array = np.expand_dims(np.zeros((6, 8)), axis=2)
for pair in pair_list:
try:
error = np.expand_dims(
np.load(
dir_base + "/" + pair + "/" + method + "/" + pose_error
),
axis=2,
)
except:
print(
"error in",
dir_base + "/" + pair + "/" + method + "/" + pose_error,
)
continue
error_array = np.concatenate((error_array, error), axis=2)
ratio_result = ratio(error_array[:, :, 1::].flatten())
f.write(
method
+ "_"
+ pose_error[11:-4]
+ " "
+ " ".join([str(i) for i in ratio_result])
+ "\n"
)
scene = "Outdoor"
dir_base = "result_errors/Outdoor/"
save_pt = "resultfinal_errors/Outdoor/"
subprocess.check_output(["mkdir", "-p", save_pt])
with open(save_pt + "ratio_methods_" + scene + ".txt", "w") as f:
f.write("5deg 10deg" + "\n")
pair_list = os.listdir(dir_base)
enhancer = os.listdir(dir_base + "/pair9/")
for method in enhancer:
pose_error_list = sorted(os.listdir(dir_base + "/pair9/" + method))
for pose_error in pose_error_list:
error_array = np.expand_dims(np.zeros((6, 8)), axis=2)
for pair in pair_list:
error = np.expand_dims(
np.load(
dir_base + "/" + pair + "/" + method + "/" + pose_error
),
axis=2,
)
error_array = np.concatenate((error_array, error), axis=2)
ratio_result = ratio(error_array[:, :, 1::].flatten())
f.write(
method
+ "_"
+ pose_error[11:-4]
+ " "
+ " ".join([str(i) for i in ratio_result])
+ "\n"
)