luni/cvpr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

save

Browse Source
This commit is contained in:
luni
2023-03-14 11:13:22 +00:00
parent d5bade988f
commit 5857aceecd
12 changed files with 487 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files
get_error.py
+23 -23
View File
@@ -8,11 +8,11 @@ if __name__ == '__main__':
garment_type_table = np.load('custom_data/garment_type_table_test.npy', allow_pickle=True).item()
folder = '/home/workspace/data/results/errors/iccv'
checkpoint_name = 'iccv_pyr_32_64_128_skin'
checkpoint_name_skirt = 'iccv_pyr_32_64_128_skin_skirt'
iteration = '75' + '000'
iteration_skirt : str = '155' + '000'
error_type = 'collision' # 'euclidean' 'collision' edge normal
checkpoint_name = 'iccv_pyr_32_64_128_2'
checkpoint_name_skirt = 'iccv_pyr_32_64_128_skirt_dress'
iteration = '125' + '000'
iteration_skirt : str = '175' + '000'
error_type = '-euclidean' # '-euclidean' '-collision' -edge -normal
errors = []
@@ -21,7 +21,7 @@ if __name__ == '__main__':
per_garment_errors_skirt = { 'Tshirt': [], 'Dress': [], 'Skirt': [], 'Top': [] }
# with open(os.path.join(folder, f'{checkpoint_name}-{iteration}-{error_type}.csv'), 'r') as f:
# with open(os.path.join(folder, f'{checkpoint_name}-{iteration}{error_type}.csv'), 'r') as f:
# csv_reader = csv.reader(f, delimiter=',')
# for row in csv_reader:
# per_garment_errors[garment_type_table[row[0].split('_')[0]]].append(float(row[1]))
@@ -34,7 +34,7 @@ if __name__ == '__main__':
# print('\nAvg non-skirt: ' + str(np.mean(errors)))
with open(os.path.join(folder, f'{checkpoint_name_skirt}-{iteration_skirt}-{error_type}.csv'), 'r') as f:
with open(os.path.join(folder, f'{checkpoint_name_skirt}-{iteration_skirt}{error_type}.csv'), 'r') as f:
csv_reader = csv.reader(f, delimiter=',')
for row in csv_reader:
per_garment_errors_skirt[garment_type_table[row[0].split('_')[0]]].append(float(row[1]))
@@ -47,24 +47,24 @@ if __name__ == '__main__':
print('Avg skirt: ' + str(np.mean(errors_skirt)))
# cum_avg_shirt = np.mean(per_garment_errors['Tshirt'] + per_garment_errors_skirt['Tshirt'])
# cum_avg_top = np.mean(per_garment_errors['Top'] + per_garment_errors_skirt['Top'])
cum_avg_shirt = np.mean(per_garment_errors['Tshirt'] + per_garment_errors_skirt['Tshirt'])
cum_avg_top = np.mean(per_garment_errors['Top'] + per_garment_errors_skirt['Top'])
# print('Global tshirt avg: ' + str(cum_avg_shirt))
# print('Global top avg: ' + str(cum_avg_top))
print('Global tshirt avg: ' + str(cum_avg_shirt))
print('Global top avg: ' + str(cum_avg_top))
# print('\nGlobal per garment: ')
print('\nGlobal per garment: ')
# global_errors = []
# per_garment_errors['Tshirt'].extend(per_garment_errors_skirt['Tshirt'])
# per_garment_errors['Top'].extend(per_garment_errors_skirt['Top'])
# per_garment_errors['Skirt'] = per_garment_errors_skirt['Skirt']
# per_garment_errors['Dress'] = per_garment_errors_skirt['Dress']
# for key in per_garment_errors.keys():
# avg = np.mean(per_garment_errors[key])
# global_errors.append(avg)
# print(key + ': ' + str(avg), 'samples: ' + str(len(per_garment_errors[key])))
global_errors = []
per_garment_errors['Tshirt'].extend(per_garment_errors_skirt['Tshirt'])
per_garment_errors['Top'].extend(per_garment_errors_skirt['Top'])
per_garment_errors['Skirt'] = per_garment_errors_skirt['Skirt']
per_garment_errors['Dress'] = per_garment_errors_skirt['Dress']
for key in per_garment_errors.keys():
avg = np.mean(per_garment_errors[key])
global_errors.append(avg)
print(key + ': ' + str(avg), 'samples: ' + str(len(per_garment_errors[key])))
# print('Global avg: ' + str(np.mean(global_errors)))
# print(len(global_errors))
print('Global avg: ' + str(np.mean(global_errors)))
print(len(global_errors))