Package 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
-
- Direct Known Subclasses:
YoloV8Translator.Builder
- Enclosing class:
- YoloV5Translator
public static class YoloV5Translator.Builder extends ObjectDetectionTranslator.ObjectDetectionBuilder<YoloV5Translator.Builder>
The builder forYoloV5Translator.
-
-
Field Summary
-
Fields inherited from class ai.djl.modality.cv.translator.ObjectDetectionTranslator.ObjectDetectionBuilder
applyRatio, 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 Description Builder()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description YoloV5Translatorbuild()Builds the translator.protected voidconfigPostProcess(java.util.Map<java.lang.String,?> arguments)YoloV5Translator.BuilderoptNmsThreshold(float nmsThreshold)Sets the NMS threshold.YoloV5Translator.BuilderoptOutputType(YoloV5Translator.YoloOutputType outputType)Sets theYoloOutputType.protected YoloV5Translator.Builderself()-
Methods inherited from class ai.djl.modality.cv.translator.ObjectDetectionTranslator.ObjectDetectionBuilder
getImageHeight, getImageWidth, optApplyRatio, 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
-
-
-
-
Method Detail
-
optOutputType
public YoloV5Translator.Builder optOutputType(YoloV5Translator.YoloOutputType outputType)
Sets theYoloOutputType.- Parameters:
outputType- theYoloOutputType- 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:
selfin classBaseImageTranslator.BaseBuilder<YoloV5Translator.Builder>
-
configPostProcess
protected void configPostProcess(java.util.Map<java.lang.String,?> arguments)
- Overrides:
configPostProcessin classObjectDetectionTranslator.ObjectDetectionBuilder<YoloV5Translator.Builder>
-
build
public YoloV5Translator build()
Builds the translator.- Returns:
- the new translator
-
-