save

aaa_helpers.txt

@@ -6,6 +6,8 @@ nohup python auto_encoder_curv.py --name="lhunor_ae_curv_iccv_resnet_2" > nohup/
 nohup python auto_encoder_body.py --name="lhunor_ae_body_iccv_resnet_2" > nohup/iccv/encoders/ae_body_iccv_resnet_2.out 2>&1 &
 
 nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32_res_skirt" > nohup/iccv/pyr/iccv_pyr_32_res_skirt.out 2>&1 &
+nohup python evaluate_ae_pyramid.py --name evaluate_iccv_pyr_32_res > nohup/iccv/evaluate/iccv_pyr_32_res.out 2>&1 &
+nohup python evaluate_ae_pyramid.py --name evaluate_iccv_pyr_32_res_skirt > nohup/iccv/evaluate/iccv_pyr_32_res_skirt.out 2>&1 &
 
 one final convolutional layer for smoothing, on the final coordinates
 

configs/iccv/config_abl_pyr_32_res.py

@@ -1,6 +1,6 @@
 config = dict()
 
-config['gpu'] = '5'
+config['gpu'] = '0'
 config['name'] = 'iccv_pyr_32_res'
 config['notes'] = ''
 config['tags'] = ['pyr', 'iccv', 'resnet']
@@ -32,7 +32,7 @@ config['pyramid']['final_upscale_enhancer'] = ''
 
 config['train_dir'] = '/home/workspace/local-data_127/train'
 config['valid_dir'] = '/home/workspace/local-data_127/valid'
-config['test_dir'] = '/home/workspace/data/dataset/test_ready'
+config['test_dir'] = '/home/workspace/local-data/test_ready'
 config['batch_size'] = 4
 config['is_skirt'] = False
 config['is_cape'] = False

configs/iccv/config_abl_pyr_32_res_skirt.py

@@ -1,6 +1,6 @@
 config = dict()
 
-config['gpu'] = '5'
+config['gpu'] = '0'
 config['name'] = 'iccv_pyr_32_res_skirt'
 config['notes'] = ''
 config['tags'] = ['pyr', 'iccv', 'resnet']

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)