save

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 &
 

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()):

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