save

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_res_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 = '170000'
 
-    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/data/results/iccv/' + checkpoint_name + '-' + iteration
 
 
     os.makedirs(output_folder, exist_ok=True)