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

Class PtSsdTranslator

java.lang.Object
ai.djl.modality.cv.translator.BaseImageTranslator<ai.djl.modality.cv.output.DetectedObjects>
ai.djl.modality.cv.translator.ObjectDetectionTranslator
ai.djl.pytorch.zoo.cv.objectdetection.PtSsdTranslator
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 PtSsdTranslator extends ai.djl.modality.cv.translator.ObjectDetectionTranslator
A PtSsdTranslator that post-process the NDArray into DetectedObjects with boundaries. Reference implementation: SSD.

  • Nested Class Summary

    Nested Classes
    Modifier and Type
    Class
    Description
    static class 
    PtSsdTranslator.Builder
    The builder for 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
    PtSsdTranslator(PtSsdTranslator.Builder builder)
    Creates the SSD translator from the given builder.

  • Method Summary

    Modifier and Type
    Method
    Description
    static PtSsdTranslator.Builder
    builder()
    Creates a builder to build a PtSSDTranslatorBuilder.
    static PtSsdTranslator.Builder
    builder(Map<String,?> arguments)
    Creates a builder to build a PtSSDTranslatorBuilder with specified arguments.
    void
    prepare(ai.djl.translate.TranslatorContext ctx)
    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.BaseImageTranslator

    getBatchifier, processInput

    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

    • PtSsdTranslator

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

  • Method Details

    • prepare

      public void prepare(ai.djl.translate.TranslatorContext ctx) throws Exception
      Specified by:
      prepare in interface ai.djl.translate.Translator<ai.djl.modality.cv.Image,ai.djl.modality.cv.output.DetectedObjects>
      Overrides:
      prepare in class ai.djl.modality.cv.translator.ObjectDetectionTranslator
      Throws:
      Exception

    • processOutput

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

    • builder

      public static PtSsdTranslator.Builder builder()
      Creates a builder to build a PtSSDTranslatorBuilder.
      Returns:
      a new builder

    • builder

      public static PtSsdTranslator.Builder builder(Map<String,?> arguments)
      Creates a builder to build a PtSSDTranslatorBuilder with specified arguments.
      Parameters:
      arguments - arguments to specify builder options
      Returns:
      a new builder