From c356ab18f3f7069309650ba3d397a9f01e213356 Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Mon, 30 Jan 2023 10:00:46 +0000 Subject: [PATCH] calculate errors --- aaa_helpers.txt | 6 ++++++ calculate_error.py | 4 ++-- evaluate_ae.py | 6 +++--- 3 files changed, 11 insertions(+), 5 deletions(-) diff --git a/aaa_helpers.txt b/aaa_helpers.txt index 39aaac4..556cf4d 100644 --- a/aaa_helpers.txt +++ b/aaa_helpers.txt @@ -5,6 +5,12 @@ nohup python auto_encoder_curv.py --name="lhunor_ae_curv_1_orig_curv_100_10p_r2_ nohup python cvae.py --name="lhunor_abl_all_10p_curv_10_r2" > nohup/review/abl_all_10p_curv_10_r2.out 2>&1 & nohup python cvae.py --name="lhunor_abl_all_10p_curv_25_r2" > nohup/review/abl_all_10p_curv_25_r2.out 2>&1 & nohup python cvae.py --name="lhunor_abl_all_10p_curv_100_r2" > nohup/review/abl_all_10p_curv_100_r2.out 2>&1 & +nohup python evaluate_ae.py --name="lhunor_abl_all_10p_curv_100_r2" > nohup/evaluate/lhunor_abl_all_10p_curv_100_r2.out 2>&1 & +nohup python evaluate_ae.py --name="lhunor_abl_all_10p_curv_10_r2" > nohup/evaluate/lhunor_abl_all_10p_curv_10_r2.out 2>&1 & +nohup python evaluate_ae.py --name="lhunor_abl_all_10p_curv_25_r2" > nohup/evaluate/lhunor_abl_all_10p_curv_25_r2.out 2>&1 & +nohup python calculate_error.py --name="lhunor_abl_all_10p_curv_100_r2" > nohup/error/abl_all_10p_curv_100_r2.out 2>&1 & +nohup python calculate_error.py --name="lhunor_abl_all_10p_curv_10_r2" > nohup/error/abl_all_10p_curv_10_r2.out 2>&1 & +nohup python calculate_error.py --name="lhunor_abl_all_10p_curv_25_r2" > nohup/error/abl_all_10p_curv_25_r2.out 2>&1 & diff --git a/calculate_error.py b/calculate_error.py index 61871d3..534ccd7 100644 --- a/calculate_error.py +++ b/calculate_error.py @@ -1,6 +1,6 @@ import os import sys -from configs.config_abl_all_post import config +from configs.config_abl_all_10p_curv_25_r2 import config os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu'] # import tensorflow as tf @@ -59,7 +59,7 @@ if __name__ == '__main__': folder = config['test_dir'] checkpoint_name = config['name'] - iteration = '' + iteration = '70000' F = np.load("custom_utils/smpl_faces" + skirt_modifier + ".npy").astype(np.int32) garment_type_table = np.load('custom_data/garment_type_table_test.npy', allow_pickle=True).item() diff --git a/evaluate_ae.py b/evaluate_ae.py index 28bb889..a93b24c 100644 --- a/evaluate_ae.py +++ b/evaluate_ae.py @@ -1,6 +1,6 @@ import os import sys -from configs.config_abl_all_post import config +from configs.config_abl_all_10p_curv_25_r2 import config os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu'] import tensorflow as tf @@ -42,7 +42,7 @@ from losses import collision_loss as collision_penalty from custom_utils.custom_utils import batch_to_dict, load_model, plot_3d, UV2Mesh, get_summary_html, maskFacesBatch, _maskFaces, plot_3d_collision from dataset import DatasetFactory, StaticDataset from custom_utils.smpl import SMPL -from models.cvae_model import CVAENew as CVAE +from models.cvae_model import CVAE from custom_utils.supersmpl_io import writeOBJ # from unif_metric import chamfer_and_score @@ -102,7 +102,7 @@ if __name__ == '__main__': ids = [id.split('_')[0] for id in ids] ids = set(ids) - iteration = '' + iteration = '70000' load_model(model, 'checkpoints/post/' + checkpoint_name, iteration=iteration) # load_model(model, 'checkpoints/' + 'abl_all_120k')