ai.djl.modality.cv.transform

Class ToTensor

java.lang.Object

ai.djl.modality.cv.transform.ToTensor

All Implemented Interfaces:

Transform

public class ToTensor
extends java.lang.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 and Description
ToTensor()

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

ToTensor

public ToTensor()

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