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

commit all

Browse Source
This commit is contained in:
luni
2023-01-25 12:25:59 +00:00
parent 1bf224841f
commit 7470ca9fb6
139 changed files with 82069 additions and 811 deletions
Download Patch File Download Diff File Expand all files Collapse all files
get_error.py
+26 -16
View File
@@ -7,54 +7,64 @@ 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'
checkpoint_name = 'abl_all'
iteration = '70000'
checkpoint_name = 'abl_2_pyr_32_64_128_256_512'
checkpoint_name_skirt = 'abl_2_pyr_32_64_128_256_512_skirt'
iteration = '50000'
iteration_skirt : str = '50000'
errors = dict()
errors_skirt = dict()
per_garment_errors = { 'Tshirt': [], 'Dress': [], 'Skirt': [], 'Top': [] }
per_garment_errors_skirt = { 'Tshirt': [], 'Jumpsuit': [], 'Trousers': [], 'Top': [] }
# checkpoint_name = 'abl_skin'
# checkpoint_name_skirt = checkpoint_name + '_skirt'
# iteration = '70000'
# iteration_skirt : str = iteration
errors = []
errors_skirt = []
per_garment_errors= { 'Tshirt': [], 'Jumpsuit': [], 'Trousers': [], 'Top': [] }
per_garment_errors_skirt = { 'Tshirt': [], 'Dress': [], 'Skirt': [], 'Top': [] }
with open(os.path.join(folder, checkpoint_name + '-' + iteration + '.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(row[1])
per_garment_errors[garment_type_table[row[0].split('_')[0]]].append(float(row[1]))
with open(os.path.join(folder, checkpoint_name + '_skirt' + '-' + iteration + '.csv'), 'r') as f:
csv_reader = csv.reader(os.path.join(folder, checkpoint_name + '_skirt' + '-' + iteration + '.csv'), delimiter=',')
with open(os.path.join(folder, checkpoint_name_skirt + '-' + iteration_skirt + '.csv'), 'r') as f:
csv_reader = csv.reader(f, delimiter=',')
for row in csv_reader:
per_garment_errors_skirt[garment_type_table[row[0]]].append(row[1])
per_garment_errors_skirt[garment_type_table[row[0].split('_')[0]]].append(float(row[1]))
print('Non-skirt:')
print('\nNon-skirt:')
for key in per_garment_errors.keys():
avg = np.mean(per_garment_errors[key])
errors.append(avg)
print(key + ': ' + str(avg))
print('Skirt:')
print('\nSkirt:')
for key in per_garment_errors_skirt.keys():
avg = np.mean(per_garment_errors_skirt[key])
errors_skirt.append(avg)
print(key + ': ' + str(avg))
print('Avg non-skirt: ' + str(np.mean(errors)))
print('\nAvg non-skirt: ' + str(np.mean(errors)))
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'])
print('Global tshirt avg: ' + str(cum_avg_shirt))
print('Global tshirt avg: ' + str(cum_avg_top))
print('Global top avg: ' + str(cum_avg_top))
print('Globl per garment: ')
print('\nGlobl 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))
print(key + ': ' + str(avg), 'samples: ' + str(len(per_garment_errors[key])))
print('Global avg: ' + str(np.mean(global_errors)))
print(len(global_errors))