Package ai.djl.modality.cv.transform

Class Normalize

java.lang.Object

ai.djl.modality.cv.transform.Normalize

All Implemented Interfaces:

Transform

public class Normalize
extends Object
implements Transform

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

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method	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 Details

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 Details

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