jax.image.resize#

jax.image.resize(image, shape, method, antialias=True, precision=Precision.HIGHEST)[源代码]#

图像缩放。

method 参数期望以下缩放方法之一：

ResizeMethod.NEAREST, "nearest"

最近邻插值。antialias 和 precision 的值将被忽略。

ResizeMethod.LINEAR, "linear", "bilinear", "trilinear", "triangle"

线性插值。如果 antialias 为 True，则在下采样时使用三角形滤波器。

ResizeMethod.CUBIC, "cubic", "bicubic", "tricubic"

三次插值，使用 Keys 三次核。

ResizeMethod.LANCZOS3, "lanczos3"

Lanczos 重采样，使用半径为 3 的核。

ResizeMethod.LANCZOS5, "lanczos5"

Lanczos 重采样，使用半径为 5 的核。

参数:

  • image – 一个 JAX 数组。

  • shape (core.Shape) – 输出形状，作为整数序列，长度等于 image 的维度数。请注意，resize() 不区分空间维度与批次或通道维度，因此这包括图像的所有维度。要表示批次或通道维度，只需保持形状中该元素不变。

  • method (str | ResizeMethod) – 要使用的缩放方法；可以是 ResizeMethod 实例或字符串。可用方法为：LINEAR, LANCZOS3, LANCZOS5, CUBIC。

  • antialias (bool) – 下采样时是否应使用抗锯齿滤波器？默认为 True。放大时无效。

返回:

缩放后的图像。