Package ai.djl.modality.cv.transform

Class CenterFit

java.lang.Object

ai.djl.modality.cv.transform.CenterFit

All Implemented Interfaces:

Transform

public class CenterFit
extends Object
implements Transform

A Transform that fit the size of an image.

Constructor Summary

Constructors

Constructor	Description
CenterFit(int width, int height)	Creates a CenterFit Transform that fit to the given width and height with given interpolation.

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

CenterFit

public CenterFit(int width,
 int height)

Creates a CenterFit Transform that fit to the given width and height with given interpolation.

Parameters:

width - the desired width

height - the desired height

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