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
。升采样时无效。
- 返回:
缩放和平移后的图像。