Spaces:
Running
Running
File size: 3,098 Bytes
c74a070 a80d6bb c74a070 a80d6bb c74a070 a80d6bb c74a070 a80d6bb c74a070 a80d6bb c74a070 a80d6bb c74a070 a80d6bb c74a070 a80d6bb c74a070 a80d6bb c74a070 a80d6bb c74a070 a80d6bb c74a070 a80d6bb c74a070 a80d6bb c74a070 a80d6bb c74a070 a80d6bb c74a070 a80d6bb c74a070 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 |
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"
)
|