ai.djl.modality.cv.util

Class NDImageUtils

java.lang.Object

ai.djl.modality.cv.util.NDImageUtils

public final class NDImageUtils
extends java.lang.Object

NDImageUtils is an image processing utility to load, reshape, and convert images using NDArray images.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	NDImageUtils.Flag Flag indicates the color channel options for images.

Method Summary

Modifier and Type	Method and Description
static NDArray	centerCrop(NDArray image) Crops an image to a square of size min(width, height).
static NDArray	centerCrop(NDArray image, int width, int height) Crops an image to a given width and height from the center of the image.
static NDArray	crop(NDArray image, int x, int y, int width, int height) Crops an image with a given location and size.
static NDArray	normalize(NDArray input, float[] mean, float[] std) Normalizes an image NDArray of shape CHW or NCHW with mean and standard deviation.
static NDArray	normalize(NDArray input, float mean, float std) Normalizes an image NDArray of shape CHW or NCHW with a single mean and standard deviation to apply to all channels.
static NDArray	resize(NDArray image, int size) Resizes an image to the given size.
static NDArray	resize(NDArray image, int width, int height) Resizes an image to the given width and height.
static NDArray	toTensor(NDArray image) Converts an image NDArray from preprocessing format to Neural Network format.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

resize

public static NDArray resize(NDArray image,
                             int size)

Resizes an image to the given size.

Parameters:

image - the image to resize

size - the new size to use for both height and width

Returns:

the resized NDList

resize

public static NDArray resize(NDArray image,
                             int width,
                             int height)

Resizes an image to the given width and height.

Parameters:

image - the image to resize

width - the desired width

height - the desired height

Returns:

the resized NDList

normalize

public static NDArray normalize(NDArray input,
                                float mean,
                                float std)

Normalizes an image NDArray of shape CHW or NCHW with a single mean and standard deviation to apply to all channels.

Parameters:

input - the image to normalize

mean - the mean to normalize with (for all channels)

std - the standard deviation to normalize with (for all channels)

Returns:

the normalized NDArray

See Also:

normalize(NDArray, float[], float[])

normalize

public static NDArray normalize(NDArray input,
                                float[] mean,
                                float[] std)

Normalizes an image NDArray of shape CHW or NCHW with mean and standard deviation.

Given mean (m1, ..., mn) and standard deviation (s1, ..., sn for n channels, this transform normalizes each channel of the input tensor with: output[i] = (input[i] - m1) / (s1).

Parameters:

input - the image to normalize

mean - the mean to normalize with for each channel

std - the standard deviation to normalize with for each channel

Returns:

the normalized NDArray

toTensor

public static NDArray toTensor(NDArray image)

Converts an image NDArray from preprocessing format to Neural Network format.

Converts an image NDArray of shape HWC in the range [0, 255] to a DataType.FLOAT32 tensor NDArray of shape CHW in the range [0, 1].

Parameters:

image - the image to convert

Returns:

the converted image

centerCrop

public static NDArray centerCrop(NDArray image)

Crops an image to a square of size min(width, height).

Parameters:

image - the image to crop

Returns:

the cropped image

See Also:

centerCrop(NDArray, int, int)

centerCrop

public static NDArray centerCrop(NDArray image,
                                 int width,
                                 int height)

Crops an image to a given width and height from the center of the image.

Parameters:

image - the image to crop

width - the desired width of the cropped image

height - the desired height of the cropped image

Returns:

the cropped image

crop

public static NDArray crop(NDArray image,
                           int x,
                           int y,
                           int width,
                           int height)

Crops an image with a given location and size.

Parameters:

image - the image to crop

x - the x coordinate of the top-left corner of the crop

y - the y coordinate of the top-left corner of the crop

width - the width of the cropped image

height - the height of the cropped image

Returns:

the cropped image