def sliding_window_mean(arr: napari.types.ImageData,
size: int = 1) -> napari.types.LayerDataTuple:
window_shape = (size, ) + (arr.shape[1:])
arr_windows = sliding_window_view(arr, window_shape=window_shape)
# as before, use squeeze to remove singleton axes
arr_windows_1d = np.squeeze(arr_windows, axis=tuple(range(1, arr.ndim)))
arr_summed = np.sum(arr_windows_1d, axis=1) / size
return (
arr_summed,
{
'translate': (size // 2, ) + (0, ) * (arr.ndim - 1),
'name': 'mean-window',
'colormap': 'magenta',
'blending': 'additive',
},
'image',
)
import numpy as np
import napari
import dask.array as da
from dask.array.lib.stride_tricks import sliding_window_view
from skimage import data
##############################################################################
# Part 1: using code to view a specific value.
blobs = data.binary_blobs(length=64, n_dim=3)
blobs_dask = da.from_array(blobs, chunks=(1, 64, 64))
# original shape [60, 1, 1, 5, 64, 64],
# use squeeze to remove singleton axes
blobs_dask_windows = np.squeeze(
sliding_window_view(blobs_dask, window_shape=(5, 64, 64)),
axis=(1, 2),
)
blobs_sum = np.sum(blobs_dask_windows, axis=1)
viewer = napari.view_image(blobs_sum)
napari.run()
##############################################################################
# Part 2: using magicgui to vary the slice thickness.
from magicgui import magicgui
def sliding_window_mean(
arr: napari.types.ImageData, size: int = 1
) -> napari.types.LayerDataTuple:
def test_sliding_window_errors(window_shape, axis):
arr = da.zeros((4, 3))
with pytest.raises(ValueError):
sliding_window_view(arr, window_shape, axis)