Spaces:
Running
Running
File size: 3,098 Bytes
4c12b36 437b5f6 4c12b36 437b5f6 4c12b36 437b5f6 4c12b36 437b5f6 4c12b36 437b5f6 4c12b36 437b5f6 4c12b36 437b5f6 4c12b36 437b5f6 4c12b36 437b5f6 4c12b36 437b5f6 4c12b36 437b5f6 4c12b36 437b5f6 4c12b36 437b5f6 4c12b36 437b5f6 4c12b36 437b5f6 4c12b36 437b5f6 4c12b36 |
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"
)
|