applied mask2mtpl to whitematter mask

code/mask2MTPL.py

@@ -20,7 +20,7 @@ def main():
         os.makedirs(args.output)
 
     # List all flo images in the input folder
-    flo_images = [os.path.join(args.input, image) for image in os.listdir(args.input) if image.endswith(".nii")]
+    flo_images = [os.path.join(args.input, image) for image in os.listdir(args.input) if image.endswith(".nii") or image.endswith(".gz")]
 
     # Loop through flo images and perform resampling
     for flo_image in flo_images: