Class FaceDetectionTranslator

java.lang.Object
com.redis.om.spring.vectorize.face.FaceDetectionTranslator
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 FaceDetectionTranslator extends Object implements ai.djl.translate.Translator<ai.djl.modality.cv.Image,ai.djl.modality.cv.output.DetectedObjects>
  • Constructor Summary

    Constructors
    Constructor
    Description
    FaceDetectionTranslator(double confThresh, double nmsThresh, double[] variance, int topK, int[][] scales, int[] steps)
     
  • Method Summary

    Modifier and Type
    Method
    Description
    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 java.lang.Object

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

    Methods inherited from interface ai.djl.translate.Translator

    getBatchifier, getExpansions, prepare
  • Constructor Details

    • FaceDetectionTranslator

      public FaceDetectionTranslator(double confThresh, double nmsThresh, double[] variance, int topK, int[][] scales, int[] steps)
  • 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>
    • processOutput

      public ai.djl.modality.cv.output.DetectedObjects processOutput(ai.djl.translate.TranslatorContext ctx, ai.djl.ndarray.NDList list)
      Specified by:
      processOutput in interface ai.djl.translate.PostProcessor<ai.djl.modality.cv.output.DetectedObjects>