Class PtSsdTranslator.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<PtSsdTranslator.Builder>
-
- ai.djl.pytorch.zoo.cv.objectdetection.PtSsdTranslator.Builder
-
- Enclosing class:
- PtSsdTranslator
public static class PtSsdTranslator.Builder extends ai.djl.modality.cv.translator.ObjectDetectionTranslator.ObjectDetectionBuilder<PtSsdTranslator.Builder>
The builder for SSD translator.
-
-
Constructor Summary
Constructors Constructor Description Builder()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description PtSsdTranslator
build()
Builds the translator.protected void
configPostProcess(java.util.Map<java.lang.String,?> arguments)
protected void
configPreProcess(java.util.Map<java.lang.String,?> arguments)
protected PtSsdTranslator.Builder
self()
PtSsdTranslator.Builder
setBoxes(int figSize, int[] featSize, int[] steps, int[] scale, int[][] aspectRatio)
Set the box parameter to reconstruct the anchor box.-
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
-
-
-
-
Method Detail
-
setBoxes
public PtSsdTranslator.Builder setBoxes(int figSize, int[] featSize, int[] steps, int[] scale, int[][] aspectRatio)
Set the box parameter to reconstruct the anchor box.- Parameters:
figSize
- image sizefeatSize
- feature sizesteps
- steps to create boxesscale
- scale between different level of generated boxesaspectRatio
- parameter go along with scale- Returns:
- this builder
-
self
protected PtSsdTranslator.Builder self()
- Specified by:
self
in classai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder<PtSsdTranslator.Builder>
-
configPreProcess
protected void configPreProcess(java.util.Map<java.lang.String,?> arguments)
- Overrides:
configPreProcess
in classai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder<PtSsdTranslator.Builder>
-
configPostProcess
protected void configPostProcess(java.util.Map<java.lang.String,?> arguments)
- Overrides:
configPostProcess
in classai.djl.modality.cv.translator.ObjectDetectionTranslator.ObjectDetectionBuilder<PtSsdTranslator.Builder>
-
build
public PtSsdTranslator build()
Builds the translator.- Returns:
- the new translator
-
-