diff --git a/modules/util/util.py b/modules/util/util.py index e696b079181db79409f5090da96fd8f2ef247251..1c6ed66e51953ba3ff7de1bfde76f0e48a909368 100644 --- a/modules/util/util.py +++ b/modules/util/util.py @@ -1331,4 +1331,8 @@ def downscale_2x(original, smoothing=False, samples_axis_first=False): def resample(x, y, z, x_min, x_max, y_min, y_max): - return RectBivariateSpline(x, y, z, [x_min, x_max, y_min, y_max]) + z_intrp = [] + for k in range(z.shape[0]): + z_k = z[k, :, :] + z_intrp.append(RectBivariateSpline(x, y, z_k, [x_min, x_max, y_min, y_max])) + return np.stack(z_intrp)