jax.image.scale_and_translate#

jax.image.scale_and_translate(image, shape, spatial_dims, scale, translation, method, antialias=True, precision=Precision.HIGHEST)[source]#

对图像应用缩放和平移。

通过使用与方法对应的采样方法从输入图像重新采样，生成形状为 ‘shape’ 的新图像。对于 2D 图像，此操作根据以下公式将输入图像中的位置 (x, y) 转换为输出图像中的位置

(x * scale[1] + translation[1], y * scale[0] + translation[0])

（请注意，*逆* 扭曲用于生成采样位置。）假设半中心像素，即整数位置 row, col 处的像素具有坐标 y, x = row + 0.5, col + 0.5，其他输入图像维度类似。

如果输出位置（像素）映射到输入边界之外的输入采样位置，则输出位置的值将设置为零。

method 参数期望以下调整大小方法之一

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 的维度数。

  • spatial_dims (Sequence[int]) – 长度为 K 的元组，指定应将传递的比例和平移应用于的空间维度。

  • scale – 一个 [K] 数组，其维度数与图像相同，包含要在每个维度中应用的比例。

  • translation – 一个 [K] 数组，其维度数与图像相同，包含要在每个维度中应用的平移。

  • method (str | ResizeMethod) – 要使用的调整大小方法；ResizeMethod 实例或字符串。可用方法有：LINEAR、LANCZOS3、LANCZOS5、CUBIC。

  • antialias (bool) – 下采样时是否应使用抗锯齿滤波器？默认为 True。对上采样没有影响。

返回值：

缩放和平移后的图像。