From 3bdb07313eaf7b0cc9a0f8928bb44470d0927645 Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Sat, 4 Mar 2023 18:52:06 +0000 Subject: [PATCH] save --- aaa_helpers.txt | 2 ++ configs/iccv/config_abl_pyr_32_res.py | 4 ++-- configs/iccv/config_abl_pyr_32_res_skirt.py | 2 +- evaluate_ae_pyramid.py | 10 +++++----- 4 files changed, 10 insertions(+), 8 deletions(-) diff --git a/aaa_helpers.txt b/aaa_helpers.txt index 2946fc3..c6ae353 100644 --- a/aaa_helpers.txt +++ b/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 diff --git a/configs/iccv/config_abl_pyr_32_res.py b/configs/iccv/config_abl_pyr_32_res.py index 5e4dc4c..678e47d 100644 --- a/configs/iccv/config_abl_pyr_32_res.py +++ b/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 diff --git a/configs/iccv/config_abl_pyr_32_res_skirt.py b/configs/iccv/config_abl_pyr_32_res_skirt.py index dd8b099..a2fef73 100644 --- a/configs/iccv/config_abl_pyr_32_res_skirt.py +++ b/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'] diff --git a/evaluate_ae_pyramid.py b/evaluate_ae_pyramid.py index 730e000..e29bbbf 100644 --- a/evaluate_ae_pyramid.py +++ b/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)