Commit ff5ac19376b917254fdc7526eb0d28e532b05162

Authored by tfmoraes
1 parent e53432ed

ENH: Using the right shape in memmap

Showing 1 changed file with 2 additions and 4 deletions   Show diff stats
invesalius/data/imagedata_utils.py
... ... @@ -462,15 +462,13 @@ def dcm2memmap(files, slice_size):
462 462 returns it and its related filename.
463 463 """
464 464 temp_file = tempfile.mktemp()
465   - shape = len(files), slice_size[0], slice_size[1]
  465 + shape = len(files), slice_size[1], slice_size[0]
466 466  
467 467 matrix = numpy.memmap(temp_file, mode='w+', dtype='int16', shape=shape)
468   -
  468 + dcm_reader = vtkgdcm.vtkGDCMImageReader()
469 469 for n, f in enumerate(files):
470   - dcm_reader = vtkgdcm.vtkGDCMImageReader()
471 470 dcm_reader.SetFileName(f)
472 471 dcm_reader.Update()
473   -
474 472 image = dcm_reader.GetOutput()
475 473 array = numpy_support.vtk_to_numpy(image.GetPointData().GetScalars())
476 474 array.shape = matrix.shape[1], matrix.shape[2]
... ...