Add check for existence of conformed_name directory

pcamach2 · Mar 14, 2022 · 9ec50d8 · 9ec50d8
1 parent 2715cad
commit 9ec50d8
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/FastSurferCNN/eval.py b/FastSurferCNN/eval.py
@@ -470,6 +470,11 @@ def fastsurfercnn(img_filename, save_as, use_cuda, gpu_small, logger, args):
         if not op.exists(op.join(sub_dir, 'orig')):
             makedirs(op.join(sub_dir, 'orig'))
 
+        # Check if conformed image directory exists and create it otherwise
+        sub_dir, out = op.split(args.conformed_name)
+        if not op.exists(sub_dir):
+            makedirs(sub_dir)
+
         # Save input image to standard location (will change once save_image functionality is integrated)
         input_image = nib.load(options.iname)
         if not options.iname.endswith('mgz'):