cn.smartjavaai.face.translator

类 SCRFDFaceTranslator

java.lang.Object

cn.smartjavaai.face.translator.SCRFDFaceTranslator

所有已实现的接口:

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 SCRFDFaceTranslator
extends Object
implements ai.djl.translate.Translator<ai.djl.modality.cv.Image,ai.djl.modality.cv.output.DetectedObjects>

SCRFD Translator

构造器概要

构造器

构造器和说明
SCRFDFaceTranslator(double confThresh, double nmsThresh, int topK, int[] steps, int inputWidth, int inputHeight)

方法概要

限定符和类型	方法和说明
List<ai.djl.modality.cv.output.Point>	decodeKps(ai.djl.ndarray.NDArray kpss)
ai.djl.ndarray.NDArray	distance2bbox(ai.djl.ndarray.NDArray points, ai.djl.ndarray.NDArray distance)
ai.djl.ndarray.NDArray	distance2kps(ai.djl.ndarray.NDArray points, ai.djl.ndarray.NDArray distance)
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)

从类继承的方法 java.lang.Object

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

从接口继承的方法 ai.djl.translate.Translator

batchProcessInput, batchProcessOutput, getBatchifier, getExpansions, prepare

构造器详细资料

SCRFDFaceTranslator

public SCRFDFaceTranslator(double confThresh,
                           double nmsThresh,
                           int topK,
                           int[] steps,
                           int inputWidth,
                           int inputHeight)

方法详细资料

processInput

public ai.djl.ndarray.NDList processInput(ai.djl.translate.TranslatorContext ctx,
                                          ai.djl.modality.cv.Image input)

指定者:

processInput 在接口中 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)

指定者:

processOutput 在接口中 ai.djl.translate.PostProcessor<ai.djl.modality.cv.output.DetectedObjects>

distance2bbox

public ai.djl.ndarray.NDArray distance2bbox(ai.djl.ndarray.NDArray points,
                                            ai.djl.ndarray.NDArray distance)

distance2kps

public ai.djl.ndarray.NDArray distance2kps(ai.djl.ndarray.NDArray points,
                                           ai.djl.ndarray.NDArray distance)

decodeKps

public List<ai.djl.modality.cv.output.Point> decodeKps(ai.djl.ndarray.NDArray kpss)