Package ai.djl.huggingface.translator

Class ZeroShotObjectDetectionTranslator.Builder

java.lang.Object

ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder<ZeroShotObjectDetectionTranslator.Builder>

ai.djl.huggingface.translator.ZeroShotObjectDetectionTranslator.Builder

Enclosing class:

ZeroShotObjectDetectionTranslator

public static final class ZeroShotObjectDetectionTranslator.Builder
extends ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder<ZeroShotObjectDetectionTranslator.Builder>

The builder for zero-shot classification translator.

Field Summary

Fields inherited from class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder

batchifier, flag, height, pipeline, width

Method Summary

Modifier and Type	Method	Description
ZeroShotObjectDetectionTranslator	build()	Builds the translator.
void	configure(Map<String,?> arguments)	Configures the builder with the model arguments.
ZeroShotObjectDetectionTranslator.Builder	optInt32(boolean int32)	Sets if use int32 datatype for the Translator.
ZeroShotObjectDetectionTranslator.Builder	optThreshold(float threshold)	Sets the threshold for prediction accuracy.
protected ZeroShotObjectDetectionTranslator.Builder	self()

Methods inherited from class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder

addTransform, configPostProcess, configPreProcess, optBatchifier, optFlag, setImageSize, setPipeline, validate

Methods inherited from class java.lang.Object

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

Method Details

self

protected ZeroShotObjectDetectionTranslator.Builder self()

Specified by:

self in class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder<ZeroShotObjectDetectionTranslator.Builder>

optThreshold

public ZeroShotObjectDetectionTranslator.Builder optThreshold(float threshold)

Sets the threshold for prediction accuracy.

Predictions below the threshold will be dropped.

Parameters:

threshold - the threshold for the prediction accuracy

Returns:

this builder

optInt32

public ZeroShotObjectDetectionTranslator.Builder optInt32(boolean int32)

Sets if use int32 datatype for the Translator.

Parameters:

int32 - true to include token types

Returns:

this builder

configure

public void configure(Map<String,?> arguments)

Configures the builder with the model arguments.

Parameters:

arguments - the model arguments

build

public ZeroShotObjectDetectionTranslator build()
                                        throws IOException

Builds the translator.

Returns:

the new translator

Throws:

IOException - if I/O error occurs