# HG changeset patch
# User greg
# Date 1512057286 18000
# Node ID b054adc68274b9d0ccae611314c80bf2606a16a1
# Parent  d9c4ef2ee7c6f4217f3df4cd9c0d2c0525ff2ef2
Uploaded

diff -r d9c4ef2ee7c6 -r b054adc68274 dmri.py
--- a/dmri.py	Thu Nov 30 10:51:01 2017 -0500
+++ b/dmri.py	Thu Nov 30 10:54:46 2017 -0500
@@ -62,9 +62,10 @@
 # Visualize the results using matplotlib.
 axial_middle = data.shape[2] // 2
 pyplot.subplot(1, 2, 1).set_axis_off()
-pyplot.imshow(data[:, :, axial_middle, 0].T, cmap='gray', origin='lower')
-pyplot.subplot(1, 2, 2).set_axis_off()
-pyplot.imshow(data[:, :, axial_middle, 10].T, cmap='gray', origin='lower')
+if args.drmi_dataset == 'stanford_hardi':
+    pyplot.imshow(data[:, :, axial_middle, 0].T, cmap='gray', origin='lower')
+    pyplot.subplot(1, 2, 2).set_axis_off()
+    pyplot.imshow(data[:, :, axial_middle, 10].T, cmap='gray', origin='lower')
 pyplot.savefig('data.png', bbox_inches='tight')
 shutil.move('data.png', args.output_png)
 # Load the b-values and b-vectors.