save

evaluate_ae_pyramid.py

@@ -1,6 +1,6 @@
 import os
 import sys
-from configs.iccv.config_abl_pyr_32_var import config
+from configs.iccv.config_abl_pyr_64_128_cape_f import config
 
 os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
 import tensorflow as tf
@@ -93,13 +93,15 @@ if __name__ == '__main__':
 
     F = np.load("custom_utils/smpl_faces" + skirt_modifier + ".npy").astype(np.int32)
 
-    ids = os.listdir(os.path.join(folder, 'pose_shape'))
-    ids = [id.split('_')[0] for id in ids]
-    ids = set(ids)
+    # ids = os.listdir(os.path.join(folder, 'pose_shape'))
+    # ids = [id.split('_')[0] for id in ids]
+    # ids = set(ids)
 
-    iteration = '235000'
-
-    load_model(model, 'checkpoints/iccv/' + checkpoint_name, iteration=iteration)
+    iteration = ''
+    if iteration == '':
+        iteration = load_model(model, 'checkpoints/iccv/' + checkpoint_name, return_iteration=True)
+    else:
+        load_model(model, 'checkpoints/iccv/' + checkpoint_name, iteration=iteration)
     # load_model(model, 'checkpoints/' + 'abl_all_120k')
 
     output_folder = '/home/workspace/local-data_128/results/iccv/' + checkpoint_name + '-' + iteration