calculate error for fisrt level
This commit is contained in:
+4
-4
@@ -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()):
|
||||
|
||||
Reference in New Issue
Block a user