save

calculate_error.py

@@ -1,6 +1,6 @@
 import os
 import sys
-from configs.iccv.config_abl_pyr_32_25p import config
+from configs.iccv.config_abl_pyr_32_64_25p import config
 
 os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
 # import tensorflow as tf
@@ -65,7 +65,7 @@ if __name__ == '__main__':
     garment_type_table = np.load('custom_data/garment_type_table_test.npy', allow_pickle=True).item()
 
 
-    output_folder = '/home/workspace/local-data/results/iccv/' + checkpoint_name + '-'
+    output_folder = '/home/workspace/local-data_128/results/iccv/' + checkpoint_name + '-'
     # check if there is a folder that starts with this name using glob
     folders = glob.glob(output_folder + '*')
     assert len(folders) == 1, f'There should be only one checkpoint that starts with this name, but found {len(folders)}'