Skip to content

nabu.processing.medfilt_cuda

source module nabu.processing.medfilt_cuda

Classes

  • MedianFilter — A class for performing median filter on GPU with CUDA

source class MedianFilter(shape, footprint=(3, 3), mode='reflect', threshold=None, cuda_options=None, abs_diff=False)

A class for performing median filter on GPU with CUDA

Constructor of Cuda Median Filter.

Parameters

  • shape : tuple — Shape of the array, in the format (n_rows, n_columns)

  • footprint : tuple — Size of the median filter, in the format (y, x).

  • mode : str — Boundary handling mode. Available modes are: - "reflect": cba|abcd|dcb - "nearest": aaa|abcd|ddd - "wrap": bcd|abcd|abc - "constant": 000|abcd|000

    Default is "reflect".

  • threshold : float, optional — Threshold for the "thresholded median filter". A thresholded median filter only replaces a pixel value by the median if this pixel value is greater or equal than median + threshold.

  • abs_diff : bool, optional — Whether to perform conditional threshold as abs(value - median)

Notes

Please refer to the documentation of the CudaProcessing class for the other parameters.

Methods

  • medfilt2 — Perform a median filter on an image (or batch of images).

source method MedianFilter.medfilt2(image, output=None)

Perform a median filter on an image (or batch of images).

Parameters

  • images : numpy.ndarray or pycuda.gpuarray — 2D image or 3D stack of 2D images

  • output : numpy.ndarray or pycuda.gpuarray, optional — Output of filtering. If provided, it must have the same shape as the input array.