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
ConstructorsConstructorDescriptionFaceDetectionTranslator(double confThresh, double nmsThresh, double[] variance, int topK, int[][] scales, int[] steps) -
Method Summary
Modifier and TypeMethodDescriptionai.djl.ndarray.NDListprocessInput(ai.djl.translate.TranslatorContext ctx, ai.djl.modality.cv.Image input) ai.djl.modality.cv.output.DetectedObjectsprocessOutput(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, waitMethods 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:
processInputin interfaceai.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:
processOutputin interfaceai.djl.translate.PostProcessor<ai.djl.modality.cv.output.DetectedObjects>
-