diff --git a/aaa_helpers.txt b/aaa_helpers.txt index c6ae353..ec29f23 100644 --- a/aaa_helpers.txt +++ b/aaa_helpers.txt @@ -143,7 +143,7 @@ nohup python evaluate_ae_pyramid.py --name evaluate_ae_abl_pyr_32_skirt > nohup/ nohup python evaluate_ae.py --name abl_norm_skirt_70k > nohup/evaluate/abl_norm_skirt_70k.out 2>&1 & nohup python calculate_error.py --name error_abl_2_pyr_32_64_128_skirt > nohup/error_ae_abl_skin_skirt_70k.out 2>&1 & -nohup python evaluate_ae_pyramid.py --name abl_2_pyr_32_64_100k > nohup/evaluate/abl_2_pyr_32_64_100k.out 2>&1 & +nohup python evaluate_ae_pyramid.py --name pyr_32_res_skirt > nohup/evaluate/pyr_32_res_skirt.out 2>&1 & nohup python evaluate_ae_pyramid.py --name abl_2_pyr_32_64_128_80k > nohup/evaluate/abl_2_pyr_32_64_128_80k.out 2>&1 & nohup python evaluate_ae_pyramid.py --name abl_2_pyr_32_64_128_256_512_skirt_50k > nohup/evaluate/abl_2_pyr_32_64_128_256_512_skirt_50k.out 2>&1 & diff --git a/calculate_error.py b/calculate_error.py index 5fbe5b7..b144f0e 100644 --- a/calculate_error.py +++ b/calculate_error.py @@ -1,6 +1,6 @@ import os import sys -from configs.review.config_abl_all import config +from configs.iccv.config_abl_pyr_32_res import config os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu'] # import tensorflow as tf @@ -59,21 +59,21 @@ if __name__ == '__main__': folder = config['test_dir'] checkpoint_name = config['name'] - iteration = '' + iteration = '195000' F = np.load("custom_utils/smpl_faces" + skirt_modifier + ".npy").astype(np.int32) garment_type_table = np.load('custom_data/garment_type_table_test.npy', allow_pickle=True).item() - output_folder = '/home/workspace/data/results/post/' + checkpoint_name + '-' + iteration + output_folder = '/home/workspace/data/results/iccv/' + checkpoint_name + '-' + iteration - faces_folder = '/home/workspace/data/results/support/faces' + ('_skirt' if config['is_skirt'] else '') - ids = os.listdir(faces_folder) - ids = [id.split('.')[0] for id in ids] - faces = {id : np.load(faces_folder + '/' + id + '.npy') for id in ids } + # faces_folder = '/home/workspace/data/results/support/faces' + ('_skirt' if config['is_skirt'] else '') + # ids = os.listdir(faces_folder) + # ids = [id.split('.')[0] for id in ids] + # faces = {id : np.load(faces_folder + '/' + id + '.npy') for id in ids } - masks_folder = '/home/workspace/data/dataset/test' + ('_skirt' if config['is_skirt'] else '') + '_ready' + ('_2' if config['is_skirt'] else '') + '/mesh_mask' + masks_folder = '/home/workspace/local-data/test' + ('_skirt' if config['is_skirt'] else '') + '_ready' + ('_2' if config['is_skirt'] else '') + '/mesh_mask' ids = os.listdir(masks_folder) ids = [id.split('.')[0] for id in ids] masks = {id : np.load(masks_folder + '/' + id + '.npy') for id in ids } @@ -86,7 +86,7 @@ if __name__ == '__main__': else: gt_name = 'mesh_posed' - orig_folder = '/home/workspace/data/dataset/test' + ('_skirt' if config['is_skirt'] else '') + '_ready' + ('_2' if config['is_skirt'] else '') + '/' + gt_name + orig_folder = '/home/workspace/local-data/test' + ('_skirt' if config['is_skirt'] else '') + '_ready' + ('_2' if config['is_skirt'] else '') + '/' + gt_name errors = dict() if config['is_skirt']: @@ -107,7 +107,7 @@ if __name__ == '__main__': for key in per_garment_errors.keys(): print(key + ': ' + str(np.mean(per_garment_errors[key]))) - with open('/home/workspace/data/results/errors/post/' + checkpoint_name + '-' + iteration + '.csv', mode='w') as f: + with open('/home/workspace/data/results/errors/iccv/' + checkpoint_name + '-' + iteration + '.csv', mode='w') as f: file_writer = csv.writer(f, delimiter=',', quotechar='"', quoting=csv.QUOTE_MINIMAL) for key in sorted(errors.keys()): diff --git a/configs/iccv/config_abl_pyr_32_res_skirt.py b/configs/iccv/config_abl_pyr_32_res_skirt.py index a2fef73..fa5b466 100644 --- a/configs/iccv/config_abl_pyr_32_res_skirt.py +++ b/configs/iccv/config_abl_pyr_32_res_skirt.py @@ -32,7 +32,7 @@ config['pyramid']['final_upscale_enhancer'] = '' config['train_dir'] = '/home/workspace/local-data_128/train_skirt' config['valid_dir'] = '/home/workspace/local-data_127/valid_skirt' -config['test_dir'] = '/home/workspace/local-data_128/test_skirt_ready_2' +config['test_dir'] = '/home/workspace/local-data/test_skirt_ready_2' config['batch_size'] = 4 config['is_skirt'] = True config['is_cape'] = False