mantidimaging.core.gpu.utility module

class mantidimaging.core.gpu.utility.CudaExecuter(dtype)[source]

Bases: object

median_filter(data, filter_size, mode, progress)[source]

Runs the median filter on a stack of 2D images asynchronously.

The data array with the median filter applied to it provided the GPU didn’t run out of space, otherwise it returns the unaltered input array.

Parameters:
  • data – The CPU data array containing a stack of 2D images.

  • filter_size – The filter size.

  • mode – The mode for the filter. Determines how the edge value are managed.

  • progress – An object for displaying the filter progress.

Returns:

Data with median filter applied on success, else unaltered input array

mantidimaging.core.gpu.utility.gpu_available()[source]
Returns:

True if cupy is installed AND working, False otherwise.