update

scripts/get_error.py

@@ -13,10 +13,10 @@ parser.add_argument('-g', '--gpu', type=str)
 args = vars(parser.parse_args())
 
 if args['debug']:
-    from configs.iccvw.pyr_64_128_cape_f_2 import config
+    from configs.iccvw.pyr_64_128_cape_f_2 import config_1
 else:
     config_module = importlib.import_module(args['config'])
-    config = config_module.config
+    config_1 = config_module.config
 
 
 import time
@@ -57,21 +57,21 @@ def calculate():
 
 
 if __name__ == '__main__':
-    skirt_modifier = '_skirt' if config['is_skirt'] else ''
+    skirt_modifier = '_skirt' if config_1['is_skirt'] else ''
 
-    name = config['name']
-    if not config['is_cape']:
-        frames_file = config['test_frames']
-        with open(os.path.join(config['test_dir'], frames_file), 'r') as f:
+    name = config_1['name']
+    if not config_1['is_cape']:
+        frames_file = config_1['test_frames']
+        with open(os.path.join(config_1['test_dir'], frames_file), 'r') as f:
             frames = f.readlines()
         frames = [Frame(frames.split()[0], int(frames.split()[1]), frames.split()[2]) for frames in frames]
     else:
         from data.dataset_cape import read_frames
-        frames = read_frames(config['test_dir'], shuffle=False, is_cape=True)
+        frames = read_frames(config_1['test_dir'], shuffle=False, is_cape=True)
 
     results_folder = '/data-local/127_2/results'
     if args['iteration'] == '':
-        folders = [f for f in os.listdir(results_folder) if os.path.isdir(os.path.join(results_folder, f)) and f.startswith(config['name'])]
+        folders = [f for f in os.listdir(results_folder) if os.path.isdir(os.path.join(results_folder, f)) and f.startswith(config_1['name'])]
         assert len(folders) == 1
         name = folders[0]
     else: