ai.djl.modality.cv.transform

Class Normalize

java.lang.Object

ai.djl.modality.cv.transform.Normalize

All Implemented Interfaces:

Transform

public class Normalize
extends java.lang.Object
implements Transform

A Transform that normalizes an image NDArray of shape CHW or NCHW.

Constructor Summary

Constructors

Constructor and Description
Normalize(float[] mean, float[] std) Creates a Normalize Transform that normalizes.

Method Summary

Modifier and Type	Method and Description
NDArray	transform(NDArray array) Applies the Transform to the given NDArray.

Methods inherited from class java.lang.Object

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

Constructor Detail

Normalize

public Normalize(float[] mean,
                 float[] std)

Creates a Normalize Transform that normalizes.

Parameters:

mean - the mean to normalize with for each channel

std - the standard deviation to normalize with for each channel

See Also:

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

Method Detail

transform

public NDArray transform(NDArray array)

Applies the Transform to the given NDArray.

Specified by:

transform in interface Transform

Parameters:

array - the NDArray on which the Transform is applied

Returns:

the output of the Transform