ai.djl.modality.cv.transform

Class CenterCrop

java.lang.Object

ai.djl.modality.cv.transform.CenterCrop

All Implemented Interfaces:

Transform

public class CenterCrop
extends java.lang.Object
implements Transform

A Transform that crops the center of an image.

Constructor Summary

Constructors

Constructor and Description
CenterCrop() Creates a CenterCrop Transform that crops to size min(width, height).
CenterCrop(int width, int height) Creates a CenterCrop Transform that crops the given width and height.

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

CenterCrop

public CenterCrop()

Creates a CenterCrop Transform that crops to size min(width, height).

CenterCrop

public CenterCrop(int width,
                  int height)

Creates a CenterCrop Transform that crops the given width and height.

Parameters:

width - the desired width of the cropped image

height - the desired height of the cropped image

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