def median_filter(piv, size=2): """Computes a median filter on u and v""" piv.u = mf(piv.u, footprint=disk(size)) piv.v = mf(piv.v, footprint=disk(size))