From f1f50cc03d08462944c79041a96ed9671ec62eda Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Sat, 28 Jan 2023 13:17:40 +0000 Subject: [PATCH] calculate error for fisrt level --- aaa_helpers.txt | 1 + calculate_error.py | 8 ++++---- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/aaa_helpers.txt b/aaa_helpers.txt index 0338d6d..b468502 100644 --- a/aaa_helpers.txt +++ b/aaa_helpers.txt @@ -3,6 +3,7 @@ nohup python auto_encoder_static.py --name="lhunor_ae_static_orig_10p_r1" > nohu nohup python auto_encoder_body.py --name="lhunor_ae_body_orig_10p_r1" > nohup/review/body_orig_10p_r1.out 2>&1 & nohup python cvae_pyramid.py --name="lhunor_abl_32_64_skirt_no_curv_r1" > nohup/review/abl_pyr_32_64_skirt_no_curv_r1.out 2>&1 & nohup python evaluate_ae_pyramid.py --name evaluate_ae_abl_pyr_32_no_curv_r1 > nohup/evaluate/abl_pyr_32_no_curv_r1.out 2>&1 & +nohup python calculate_error.py --name error_abl_pyr_32_no_curv_r1 > nohup/error/abl_pyr_32_no_curv_r1.out 2>&1 & one final convolutional layer for smoothing, on the final coordinates diff --git a/calculate_error.py b/calculate_error.py index 61871d3..bb3a8f5 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_pyr_32_skirt import config os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu'] # import tensorflow as tf @@ -59,13 +59,13 @@ if __name__ == '__main__': folder = config['test_dir'] checkpoint_name = config['name'] - iteration = '' + iteration = '340000' 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() - output_folder = '/home/workspace/data/results/post/' + checkpoint_name + '-' + iteration + output_folder = '/home/workspace/data/results/' + checkpoint_name + '-' + iteration faces_folder = '/home/workspace/data/results/support/faces' + ('_skirt' if config['is_skirt'] else '') @@ -107,7 +107,7 @@ if __name__ == '__main__': for key in per_garment_errors.keys(): print(key + ': ' + str(np.mean(per_garment_errors[key]))) - with open('/home/workspace/data/results/errors/post/' + checkpoint_name + '-' + iteration + '.csv', mode='w') as f: + with open('/home/workspace/data/results/errors/' + checkpoint_name + '-' + iteration + '.csv', mode='w') as f: file_writer = csv.writer(f, delimiter=',', quotechar='"', quoting=csv.QUOTE_MINIMAL) for key in sorted(errors.keys()):