Package ai.djl.pytorch.zoo.cv.objectdetection

Class 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.

    • 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
      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

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

        addTransform, 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

      • 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 size
        featSize - feature size
        steps - steps to create boxes
        scale - scale between different level of generated boxes
        aspectRatio - parameter go along with scale
        Returns:
        this builder

      • configPreProcess

        protected void configPreProcess​(java.util.Map<java.lang.String,​?> arguments)
        Overrides:
        configPreProcess in class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder<PtSsdTranslator.Builder>

      • configPostProcess

        protected void configPostProcess​(java.util.Map<java.lang.String,​?> arguments)
        Overrides:
        configPostProcess in class ai.djl.modality.cv.translator.ObjectDetectionTranslator.ObjectDetectionBuilder<PtSsdTranslator.Builder>

      • build

        public PtSsdTranslator build()
        Builds the translator.
        Returns:
        the new translator