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

Class TfSsdTranslator

java.lang.Object
ai.djl.modality.cv.translator.BaseImageTranslator<ai.djl.modality.cv.output.DetectedObjects>
ai.djl.modality.cv.translator.ObjectDetectionTranslator
ai.djl.tensorflow.zoo.cv.objectdetction.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

    applyRatio, classes, removePadding, threshold

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

    height, pipeline, width

  • Constructor Summary

    Constructors
    Modifier
    Constructor
    Description
    protected
    TfSsdTranslator(TfSsdTranslator.Builder builder)
    Creates the SSD translator from the given builder.

  • Method Summary

    Modifier and Type
    Method
    Description
    static TfSsdTranslator.Builder
    builder()
    Creates a builder to build a TfSSDTranslator.
    static TfSsdTranslator.Builder
    builder(Map<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

    Methods inherited from interface ai.djl.translate.Translator

    batchProcessInput, batchProcessOutput, getExpansions

  • Constructor Details

    • TfSsdTranslator

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

  • Method Details

    • 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(Map<String,?> arguments)
      Creates a builder to build a TfSSDTranslator with specified arguments.
      Parameters:
      arguments - arguments to specify builder options
      Returns:
      a new builder