save

calculate_error.py

@@ -1,6 +1,6 @@
 import os
 import sys
-from configs.iccv.config_abl_pyr_32_64_128_skin_skirt import config
+from configs.iccv.config_abl_pyr_32_64_128_tshirt import config
 
 os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
 import tensorflow as tf
@@ -58,7 +58,7 @@ if __name__ == '__main__':
     # smpl = SMPL(use_one_hot=config['use_one_hot'])
     skirt_modifier = '_skirt' if config['is_skirt'] else ''
 
-    error_type = 'collision' # 'euclidean' 'collision' edge normal
+    error_type = 'euclidean' # 'euclidean' 'collision' edge normal
 
     folder = config['test_dir']
 
@@ -92,9 +92,16 @@ if __name__ == '__main__':
     frames = [frame for frame in frames if not frame.endswith('_fixed_coll.npy')]
     # frames = frames[:100]
 
-    print('Reading bodies...')
-    bodies_folder = '/home/workspace/local-data' + ('' if config['is_skirt'] else '_127') + '/test' + ('_skirt' if config['is_skirt'] else '') + '_ready' + ('_2' if config['is_skirt'] else '') + '/mesh_posed_body_no_orient'
-    bodies = {frame : np.load(bodies_folder + '/' + frame[:-1] + 'z')['mesh_posed_body_no_orient'] for frame in tqdm(frames) }
+    # if 'frames_file_test' in config.keys():
+    #     with open(os.path.join(folder, config['frames_file_test']), 'r') as f:
+    #             lines = f.readlines()
+    #             lines = [line.strip() for line in lines]
+    #     frames = lines
+
+    if error_type == 'collision':
+        print('Reading bodies...')
+        bodies_folder = '/home/workspace/local-data' + ('' if config['is_skirt'] else '_127') + '/test' + ('_skirt' if config['is_skirt'] else '') + '_ready' + ('_2' if config['is_skirt'] else '') + '/mesh_posed_body_no_orient'
+        bodies = {frame : np.load(bodies_folder + '/' + frame[:-1] + 'z')['mesh_posed_body_no_orient'] for frame in tqdm(frames) }
 
     if config['remove_global_orientation']:
         gt_name = 'mesh_posed_no_orient'