commit all

evaluate_ae_pyramid.py

@@ -1,6 +1,6 @@
 import os
 import sys
-from configs.config_abl_pyr_32 import config
+from configs.config_abl_2_pyr_32_64_128_256_512 import config
 
 os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
 import tensorflow as tf
@@ -80,7 +80,6 @@ if __name__ == '__main__':
     parser.add_argument('--name', type=str)
     args = vars(parser.parse_args())
 
-    smpl = SMPL()
     skirt_modifier = '_skirt' if config['is_skirt'] else ''
 
     folder = config['test_dir']
@@ -98,7 +97,7 @@ if __name__ == '__main__':
     ids = [id.split('_')[0] for id in ids]
     ids = set(ids)
 
-    iteration = '430000'
+    iteration = '50000'
 
     load_model(model, 'checkpoints/' + checkpoint_name, iteration=iteration)
     # load_model(model, 'checkpoints/' + 'abl_all_120k')