lot more training

evaluate_ae_pyramid.py

@@ -1,6 +1,6 @@
 import os
 import sys
-from configs.config_abl_2_pyr_32_64_128_256_512 import config
+from configs.iccv.config_abl_pyr_32_reg_skirt import config
 
 os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
 import tensorflow as tf
@@ -27,7 +27,7 @@ import numpy as np
 import plotly.graph_objects as go
 import plotly.express as ex
 from PIL import Image
-from models.pyramid_model_optimized import Pyramid
+from models.pyramid_model_optimized import PyramidNew as Pyramid
 from postprocessing import fix_collision
 
 if config['is_skirt']:
@@ -97,12 +97,12 @@ if __name__ == '__main__':
     ids = [id.split('_')[0] for id in ids]
     ids = set(ids)
 
-    iteration = '50000'
+    iteration = '190000'
 
-    load_model(model, 'checkpoints/' + checkpoint_name, iteration=iteration)
+    load_model(model, 'checkpoints/iccv/' + checkpoint_name, iteration=iteration)
     # load_model(model, 'checkpoints/' + 'abl_all_120k')
 
-    output_folder = '/home/workspace/data/results/' + checkpoint_name + '-' + iteration
+    output_folder = '/home/workspace/local-data_128/results/iccv/' + checkpoint_name + '-' + iteration
 
 
     os.makedirs(output_folder, exist_ok=True)
@@ -128,10 +128,10 @@ if __name__ == '__main__':
                                         resolutions=dataset_resolutions, 
                                         in_memory=True)
 
-    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/local-data_128/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 }
 
     for batch in tqdm.tqdm(dataset_test):
         batch = static_dataset_test.expand_batch(batch_to_dict(batch, config['signature_dynamic']))