Package ai.djl.modality.cv.translator

Class YoloWorldTranslator.Builder

java.lang.Object

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

ai.djl.modality.cv.translator.YoloWorldTranslator.Builder

Enclosing class:

YoloWorldTranslator

public static class YoloWorldTranslator.Builder
extends BaseImageTranslator.BaseBuilder<YoloWorldTranslator.Builder>

The builder for YoloWorldTranslator.

Field Summary

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

batchifier, flag, height, pipeline, width

Constructor Summary

Constructors

Constructor	Description
Builder()

Method Summary

Modifier and Type	Method	Description
YoloWorldTranslator	build()	Builds the translator.
protected void	configPostProcess(Map<String,?> arguments)
YoloWorldTranslator.Builder	optClipModelPath(String clipModelPath)	Sets the clip model file path, default value "clip.pt".
YoloWorldTranslator.Builder	optNmsThreshold(float nmsThreshold)	Sets the NMS threshold.
YoloWorldTranslator.Builder	optThreshold(float threshold)	Sets the threshold for prediction accuracy.
protected YoloWorldTranslator.Builder	self()

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

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

Methods inherited from class java.lang.Object

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

Constructor Details

Builder

public Builder()

Method Details

self

protected YoloWorldTranslator.Builder self()

Specified by:

self in class BaseImageTranslator.BaseBuilder<YoloWorldTranslator.Builder>

optThreshold

public YoloWorldTranslator.Builder optThreshold(float threshold)

Sets the threshold for prediction accuracy.

Predictions below the threshold will be dropped.

Parameters:

threshold - the threshold for prediction accuracy

Returns:

the builder

optNmsThreshold

public YoloWorldTranslator.Builder optNmsThreshold(float nmsThreshold)

Sets the NMS threshold.

Parameters:

nmsThreshold - the NMS threshold

Returns:

this builder

optClipModelPath

public YoloWorldTranslator.Builder optClipModelPath(String clipModelPath)

Sets the clip model file path, default value "clip.pt".

Parameters:

clipModelPath - the clip model file path

Returns:

this builder

configPostProcess

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

Overrides:

configPostProcess in class BaseImageTranslator.BaseBuilder<YoloWorldTranslator.Builder>

build

public YoloWorldTranslator build()

Builds the translator.

Returns:

the new translator