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

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

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

    • Constructor Detail

      • Builder

        public Builder()

    • Method Detail

      • optNumDetectionsOutputName

        public TfSsdTranslator.Builder optNumDetectionsOutputName​(java.lang.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​(java.lang.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​(java.lang.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​(java.lang.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

      • configPreProcess

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

      • configPostProcess

        protected void configPostProcess​(java.util.Map<java.lang.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