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

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.

  • Field Summary

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

    applyRatio, removePadding, 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

    Modifier and Type
    Method
    Description
    PtSsdTranslator
    build()
    Builds the translator.
    protected void
    configPostProcess(Map<String,?> arguments)
    protected void
    configPreProcess(Map<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

    optApplyRatio, 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, setImageSize, setPipeline

    Methods inherited from class java.lang.Object

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

  • Constructor Details

    • Builder

      public Builder()

  • Method Details

    • 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

    • self

      protected PtSsdTranslator.Builder self()
      Specified by:
      self in class ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder<PtSsdTranslator.Builder>

    • configPreProcess

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

    • configPostProcess

      protected void configPostProcess(Map<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