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)[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.