ai.djl.modality.cv.transform

Class RandomColorJitter

java.lang.Object

ai.djl.modality.cv.transform.RandomColorJitter

All Implemented Interfaces:

Transform

public class RandomColorJitter
extends java.lang.Object
implements Transform

A Transform that randomly jitters the brightness, contrast, saturation, and hue of an image.

Constructor Summary

Constructors

Constructor and Description
RandomColorJitter(float brightness, float contrast, float saturation, float hue) Creates a RandomColorJitter Transform.

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

RandomColorJitter

public RandomColorJitter(float brightness,
                         float contrast,
                         float saturation,
                         float hue)

Creates a RandomColorJitter Transform.

Parameters:

brightness - the brightness factor from 0 to 1

contrast - the contrast factor from 0 to 1

saturation - the saturation factor from 0 to 1

hue - the hue factor from 0 to 1

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