Package ai.djl.modality.cv.translator

Class BaseImageTranslator.BaseBuilder<T extends BaseImageTranslator.BaseBuilder>

java.lang.Object

ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder<T>

Type Parameters:

T - the concrete builder type

Direct Known Subclasses:

BaseImageTranslator.ClassificationBuilder, ImageFeatureExtractor.Builder, SimplePoseTranslator.Builder, YoloPoseTranslator.Builder

Enclosing class:

BaseImageTranslator<T>

public abstract static class BaseImageTranslator.BaseBuilder<T extends BaseImageTranslator.BaseBuilder>
extends Object

A builder to extend for all classes extending the BaseImageTranslator.

Field Summary

Fields

Modifier and Type	Field	Description
protected Batchifier	batchifier
protected Image.Flag	flag
protected int	height
protected Pipeline	pipeline
protected int	width

Constructor Summary

Constructors

Constructor	Description
BaseBuilder()

Method Summary

Modifier and Type	Method	Description
T	addTransform(Transform transform)	Adds the Transform to the Pipeline use for pre-processing the image.
protected void	configPostProcess(Map<String,?> arguments)
protected void	configPreProcess(Map<String,?> arguments)
T	optBatchifier(Batchifier batchifier)	Sets the Batchifier for the Translator.
T	optFlag(Image.Flag flag)	Sets the optional Image.Flag (default is Image.Flag.COLOR).
protected abstract T	self()
T	setPipeline(Pipeline pipeline)	Sets the Pipeline to use for pre-processing the image.
protected void	validate()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Details

width

protected int width

height

protected int height

flag

protected Image.Flag flag

pipeline

protected Pipeline pipeline

batchifier

protected Batchifier batchifier

Constructor Details

BaseBuilder

public BaseBuilder()

Method Details

optFlag

public T optFlag(Image.Flag flag)

Sets the optional Image.Flag (default is Image.Flag.COLOR).

Parameters:

flag - the color mode for the images

Returns:

this builder

setPipeline

public T setPipeline(Pipeline pipeline)

Sets the Pipeline to use for pre-processing the image.

Parameters:

pipeline - the pre-processing pipeline

Returns:

this builder

addTransform

public T addTransform(Transform transform)

Adds the Transform to the Pipeline use for pre-processing the image.

Parameters:

transform - the Transform to be added

Returns:

this builder

optBatchifier

public T optBatchifier(Batchifier batchifier)

Sets the Batchifier for the Translator.

Parameters:

batchifier - the Batchifier to be set

Returns:

this builder

self

protected abstract T self()

validate

protected void validate()

configPreProcess

protected void configPreProcess(Map<String,?> arguments)

configPostProcess

protected void configPostProcess(Map<String,?> arguments)