ai.djl.modality.cv.translator

Class YoloV5Translator.Builder

java.lang.Object

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

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

ai.djl.modality.cv.translator.ObjectDetectionTranslator.ObjectDetectionBuilder<YoloV5Translator.Builder>

ai.djl.modality.cv.translator.YoloV5Translator.Builder

Enclosing class:

YoloV5Translator

public static class YoloV5Translator.Builder
extends ObjectDetectionTranslator.ObjectDetectionBuilder<YoloV5Translator.Builder>

The builder for YoloV5Translator.

Field Summary

Fields inherited from class ai.djl.modality.cv.translator.ObjectDetectionTranslator.ObjectDetectionBuilder

imageHeight, imageWidth, threshold

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

synsetLoader

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

batchifier, flag, height, pipeline, width

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
YoloV5Translator	build() Builds the translator.
protected void	configPostProcess(java.util.Map<java.lang.String,?> arguments)
YoloV5Translator.Builder	optNmsThreshold(float nmsThreshold) Sets the NMS threshold.
YoloV5Translator.Builder	optOutputType(YoloV5Translator.YoloOutputType outputType) Sets the YoloOutputType.
protected YoloV5Translator.Builder	self()

Methods inherited from class ai.djl.modality.cv.translator.ObjectDetectionTranslator.ObjectDetectionBuilder

getImageHeight, getImageWidth, optRescaleSize, optThreshold

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

optSynset, optSynsetArtifactName, optSynsetUrl, validate

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

addTransform, configPreProcess, optBatchifier, optFlag, setPipeline

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder()

Method Detail

optOutputType

public YoloV5Translator.Builder optOutputType(YoloV5Translator.YoloOutputType outputType)

Sets the YoloOutputType.

Parameters:

outputType - the YoloOutputType

Returns:

this builder

optNmsThreshold

public YoloV5Translator.Builder optNmsThreshold(float nmsThreshold)

Sets the NMS threshold.

Parameters:

nmsThreshold - the NMS threshold

Returns:

this builder

self

protected YoloV5Translator.Builder self()

Specified by:

self in class BaseImageTranslator.BaseBuilder<YoloV5Translator.Builder>

configPostProcess

protected void configPostProcess(java.util.Map<java.lang.String,?> arguments)

Overrides:

configPostProcess in class ObjectDetectionTranslator.ObjectDetectionBuilder<YoloV5Translator.Builder>

build

public YoloV5Translator build()

Builds the translator.

Returns:

the new translator