update todo

m-albert · Oct 23, 2020 · 603016d · 603016d
1 parent d889c21
commit 603016d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/TODO.md b/TODO.md
@@ -8,9 +8,8 @@
 
 
 ###Performance
-- stream fused blocks directly to file
 - perform blockwise transformations to handle large target volumes
-- optimize weight calculation (downscaling seems to take long)
+- optimize weight calculation (downscaling seems to take long) (dask.array.coarsen?)
 
 
 ### Registration

diff --git a/mvregfus/multiview.py b/mvregfus/multiview.py
@@ -1686,8 +1686,9 @@ def affine_transform_dask(
 
  def transform_chunk(x, matrix, offset, input, kwargs, block_info=None):
 
- N = x.ndim
+ N = input.ndim
  input_shape = input.shape
+ chunk_shape = block_info[None]['chunk-shape']
 
  chunk_offset = np.array([i[0] for i in block_info[0]['array-location']])
  # print('chunk_offset', chunk_offset)
@@ -1745,7 +1746,7 @@ def transform_chunk(x, matrix, offset, input, kwargs, block_info=None):
  transformed_chunk = curr_ndimage.affine_transform(asarray(input_relevant),
  asarray(matrix),
  asarray(offset_modified),
- output_shape=x.shape,
+ output_shape=chunk_shape,
  order = 1)
  # **kwargs)