Optimize median filtering

Median filtering from skimage is very [memory intensive](https://github.com/scipy/scipy/issues/4878) as it is using scipy's `ndi.median_filter()` under the hood. This can cause a `MemoryError` when the kernel or the image is too large. See whether we can implement our own version based on some of the suggestions in the linked SciPy issue.