Package ai.djl.modality.cv.transform

Class ToTensor

java.lang.Object

ai.djl.modality.cv.transform.ToTensor

All Implemented Interfaces:

Transform

public class ToTensor
extends Object
implements Transform

A Transform that converts an image NDArray from preprocessing format to Neural Network format.

See Also:

• NDImageUtils.toTensor(NDArray)

Constructor Summary

Constructors

Constructor	Description
ToTensor()

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

ToTensor

public ToTensor()

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