jax.image module#

Image manipulation functions.

More image manipulation functions can be found in libraries built on top of JAX, such as PIX.

Image manipulation functions#

resize(image, shape, method[, antialias, ...])

Image resize.

scale_and_translate(image, shape, ...[, ...])

Apply a scale and translation to an image.

Argument classes#

class jax.image.ResizeMethod(value, names=None, *, module=None, qualname=None, type=None, start=1, boundary=None)[source]#

Image resize method.

Possible values are:

NEAREST:

Nearest-neighbor interpolation.

LINEAR:

Linear interpolation.

LANCZOS3:

Lanczos resampling, using a kernel of radius 3.

LANCZOS5:

Lanczos resampling, using a kernel of radius 5.

CUBIC:

Cubic interpolation, using the Keys cubic kernel.