Spaces:
Running
Running
| 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") | |