# HG changeset patch
# User greg
# Date 1509885832 18000
# Node ID 0c3113fdfae6d0009aa02fff43ccd69941373652
# Parent  814082359f36a6802daf70429de185991b7e44fb
Uploaded

diff -r 814082359f36 -r 0c3113fdfae6 dmri.py
--- a/dmri.py	Sun Nov 05 07:35:51 2017 -0500
+++ b/dmri.py	Sun Nov 05 07:43:52 2017 -0500
@@ -20,11 +20,14 @@
 input_dir = args.drmi_dataset
 if args.drmi_dataset == 'sherbrooke_3shell':
     fetch_sherbrooke_3shell()
+    fdwi = os.path.join(input_dir, 'HARDI193.nii.gz')
+    fbval = os.path.join(input_dir, 'HARDI193.bval')
+    fbvec = os.path.join(input_dir, 'HARDI193.bvec')
 elif args.drmi_dataset == 'stanford_hardi':
     fetch_stanford_hardi()
-fdwi = os.path.join(input_dir, 'HARDI193.nii.gz')
-fbval = os.path.join(input_dir, 'HARDI193.bval')
-fbvec = os.path.join(input_dir, 'HARDI193.bvec')
+    fdwi = os.path.join(input_dir, 'HARDI150.nii.gz')
+    fbval = os.path.join(input_dir, 'HARDI150.bval')
+    fbvec = os.path.join(input_dir, 'HARDI150.bvec')
 # Load the dMRI datasets.
 img = nibabel.load(fdwi)
 data = img.get_data()