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
- 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
FieldsModifier and TypeFieldDescriptionprotected Batchifier
protected Image.Flag
protected int
protected Pipeline
protected int
-
Constructor Summary
Constructors -
Method Summary
Modifier and TypeMethodDescriptionaddTransform
(Transform transform) protected void
configPostProcess
(Map<String, ?> arguments) protected void
configPreProcess
(Map<String, ?> arguments) optBatchifier
(Batchifier batchifier) Sets theBatchifier
for theTranslator
.optFlag
(Image.Flag flag) Sets the optionalImage.Flag
(default isImage.Flag.COLOR
).protected abstract T
self()
setPipeline
(Pipeline pipeline) Sets thePipeline
to use for pre-processing the image.protected void
validate()
-
Field Details
-
width
protected int width -
height
protected int height -
flag
-
pipeline
-
batchifier
-
-
Constructor Details
-
BaseBuilder
public BaseBuilder()
-
-
Method Details
-
optFlag
Sets the optionalImage.Flag
(default isImage.Flag.COLOR
).- Parameters:
flag
- the color mode for the images- Returns:
- this builder
-
setPipeline
Sets thePipeline
to use for pre-processing the image.- Parameters:
pipeline
- the pre-processing pipeline- Returns:
- this builder
-
addTransform
- Parameters:
transform
- theTransform
to be added- Returns:
- this builder
-
optBatchifier
Sets theBatchifier
for theTranslator
.- Parameters:
batchifier
- theBatchifier
to be set- Returns:
- this builder
-
self
-
validate
protected void validate() -
configPreProcess
-
configPostProcess
-