Package ai.djl.tensorflow.zoo.cv.objectdetction

Class TfSsdTranslator.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<TfSsdTranslator.Builder>
ai.djl.tensorflow.zoo.cv.objectdetction.TfSsdTranslator.Builder
Enclosing class:
TfSsdTranslator
public static class TfSsdTranslator.Builder extends ai.djl.modality.cv.translator.ObjectDetectionTranslator.ObjectDetectionBuilder<TfSsdTranslator.Builder>
The builder for TensorFlow SSD translator.

  • Constructor Details

    • Builder

      public Builder()

  • Method Details

    • optNumDetectionsOutputName

      public TfSsdTranslator.Builder optNumDetectionsOutputName(String numDetectionsOutputName)
      Set the output name used for number of detections.

      You can find the output names of TensorFlow models by calling `model.describeOutput()` after loading it.

      Parameters:
      numDetectionsOutputName - output name for number of detections
      Returns:
      this builder

    • optBoundingBoxOutputName

      public TfSsdTranslator.Builder optBoundingBoxOutputName(String boundingBoxOutputName)
      Set the output name used for bounding boxes. You can find the output names of TensorFlow models by calling `model.describeOutput()` after loading it.
      Parameters:
      boundingBoxOutputName - output name for bounding boxes
      Returns:
      this builder

    • optScoresOutputName

      public TfSsdTranslator.Builder optScoresOutputName(String scoresOutputName)
      Set the output name used for detection scores. You can find the output names of TensorFlow models by calling `model.describeOutput()` after loading it.
      Parameters:
      scoresOutputName - output name for detection scores
      Returns:
      this builder

    • optClassLabelOutputName

      public TfSsdTranslator.Builder optClassLabelOutputName(String classLabelOutputName)
      Set the output name used for class label. You can find the output names of TensorFlow models by calling `model.describeOutput()` after loading it.
      Parameters:
      classLabelOutputName - output name for class label
      Returns:
      this builder

    • optMaxBoxes

      public TfSsdTranslator.Builder optMaxBoxes(int maxBoxes)
      Set the maximum number of bounding boxes to display.
      Parameters:
      maxBoxes - maximum number of bounding boxes to display
      Returns:
      this builder

    • self

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

    • configPreProcess

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

    • configPostProcess

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

    • build

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