Class ImageFeatureExtractor

java.lang.Object
ai.djl.modality.cv.translator.BaseImageTranslator<byte[]>
ai.djl.modality.cv.translator.ImageFeatureExtractor
All Implemented Interfaces:
PostProcessor<byte[]>, PreProcessor<Image>, Translator<Image,byte[]>

public class ImageFeatureExtractor extends BaseImageTranslator<byte[]>
A generic Translator for Image Classification feature extraction tasks.
  • Method Details

    • processOutput

      public byte[] processOutput(TranslatorContext ctx, NDList list)
      Processes the output NDList to the corresponding output object.
      Parameters:
      ctx - the toolkit used for post-processing
      list - the output NDList after inference, usually immutable in engines like PyTorch. @see Issue 1774
      Returns:
      the output object of expected type
    • builder

      public static ImageFeatureExtractor.Builder builder()
      Creates a builder to build a ImageFeatureExtractor.
      Returns:
      a new builder
    • builder

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