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

Class TfSsdTranslator

  • All Implemented Interfaces:
    ai.djl.translate.PostProcessor<ai.djl.modality.cv.output.DetectedObjects>, ai.djl.translate.PreProcessor<ai.djl.modality.cv.Image>, ai.djl.translate.Translator<ai.djl.modality.cv.Image,​ai.djl.modality.cv.output.DetectedObjects>
    public class TfSsdTranslator
extends ai.djl.modality.cv.translator.ObjectDetectionTranslator
    A TfSsdTranslator that post-process the NDArray into DetectedObjects with boundaries. Reference implementation: SSD.

    • Nested Class Summary

      Nested Classes 
      Modifier and Type Class Description
      static class  TfSsdTranslator.Builder
      The builder for TensorFlow SSD translator.

      • Nested classes/interfaces inherited from class ai.djl.modality.cv.translator.ObjectDetectionTranslator

        ai.djl.modality.cv.translator.ObjectDetectionTranslator.ObjectDetectionBuilder<T extends ai.djl.modality.cv.translator.ObjectDetectionTranslator.ObjectDetectionBuilder>

      • Nested classes/interfaces inherited from class ai.djl.modality.cv.translator.BaseImageTranslator

        ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder<T extends ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder>, ai.djl.modality.cv.translator.BaseImageTranslator.ClassificationBuilder<T extends ai.djl.modality.cv.translator.BaseImageTranslator.BaseBuilder>, ai.djl.modality.cv.translator.BaseImageTranslator.SynsetLoader

    • Field Summary

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

        classes, imageHeight, imageWidth, threshold

      • Fields inherited from class ai.djl.modality.cv.translator.BaseImageTranslator

        pipeline

    • Method Summary

      All Methods Static Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      static TfSsdTranslator.Builder builder()
      Creates a builder to build a TfSSDTranslator.
      static TfSsdTranslator.Builder builder​(java.util.Map<java.lang.String,​?> arguments)
      Creates a builder to build a TfSSDTranslator with specified arguments.
      ai.djl.ndarray.NDList processInput​(ai.djl.translate.TranslatorContext ctx, ai.djl.modality.cv.Image input)
      ai.djl.modality.cv.output.DetectedObjects processOutput​(ai.djl.translate.TranslatorContext ctx, ai.djl.ndarray.NDList list)

      • Methods inherited from class ai.djl.modality.cv.translator.ObjectDetectionTranslator

        prepare

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

        getBatchifier

      • Methods inherited from class java.lang.Object

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

    • Constructor Detail

      • TfSsdTranslator

        protected TfSsdTranslator​(TfSsdTranslator.Builder builder)
        Creates the SSD translator from the given builder.
        Parameters:
        builder - the builder for the translator

    • Method Detail

      • processInput

        public ai.djl.ndarray.NDList processInput​(ai.djl.translate.TranslatorContext ctx,
                                          ai.djl.modality.cv.Image input)
        Specified by:
        processInput in interface ai.djl.translate.PreProcessor<ai.djl.modality.cv.Image>
        Overrides:
        processInput in class ai.djl.modality.cv.translator.BaseImageTranslator<ai.djl.modality.cv.output.DetectedObjects>

      • processOutput

        public ai.djl.modality.cv.output.DetectedObjects processOutput​(ai.djl.translate.TranslatorContext ctx,
                                                               ai.djl.ndarray.NDList list)

      • builder

        public static TfSsdTranslator.Builder builder()
        Creates a builder to build a TfSSDTranslator.
        Returns:
        a new builder

      • builder

        public static TfSsdTranslator.Builder builder​(java.util.Map<java.lang.String,​?> arguments)
        Creates a builder to build a TfSSDTranslator with specified arguments.
        Parameters:
        arguments - arguments to specify builder options
        Returns:
        a new builder