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
-
- 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
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 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 theYoloOutputType
.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
-
-
-
-
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:
self
in classBaseImageTranslator.BaseBuilder<YoloV5Translator.Builder>
-
configPostProcess
protected void configPostProcess(java.util.Map<java.lang.String,?> arguments)
- Overrides:
configPostProcess
in classObjectDetectionTranslator.ObjectDetectionBuilder<YoloV5Translator.Builder>
-
build
public YoloV5Translator build()
Builds the translator.- Returns:
- the new translator
-
-