ai.djl.modality.cv.translator

Class ImageTranslator.BaseBuilder<T extends ImageTranslator.BaseBuilder>

java.lang.Object

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

Type Parameters:

T - the concrete builder type

Direct Known Subclasses:

ImageClassificationTranslator.Builder, InstanceSegmentationTranslator.Builder, SimplePoseTranslator.Builder, SingleShotDetectionTranslator.Builder

Enclosing class:

ImageTranslator<T>

public abstract static class ImageTranslator.BaseBuilder<T extends ImageTranslator.BaseBuilder>
extends java.lang.Object

A builder to extend for all classes extending the ImageTranslator.

Field Summary

Fields

Modifier and Type	Field and Description
protected NDImageUtils.Flag	flag
protected Pipeline	pipeline

Constructor Summary

Constructors

Constructor and Description
BaseBuilder()

Method Summary

Modifier and Type	Method and Description
T	optFlag(NDImageUtils.Flag flag) Sets the optional NDImageUtils.Flag (default is NDImageUtils.Flag#COLOR).
protected abstract T	self()
T	setPipeline(Pipeline pipeline) Sets the Pipeline to use for pre-processing the image.

Methods inherited from class java.lang.Object

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

Field Detail

flag

protected NDImageUtils.Flag flag

pipeline

protected Pipeline pipeline

Constructor Detail

BaseBuilder

public BaseBuilder()

Method Detail

optFlag

public T optFlag(NDImageUtils.Flag flag)

Sets the optional NDImageUtils.Flag (default is NDImageUtils.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

self

protected abstract T self()