cn.smartjavaai.face.model.facedect.mtcnn

类 ONetModel

java.lang.Object

cn.smartjavaai.face.model.facedect.mtcnn.ONetModel

public class ONetModel
extends Object

作者:

dwj

构造器概要

构造器

构造器和说明
ONetModel()

方法概要

限定符和类型	方法和说明
static MtcnnBatchResult	processBatchBoxes(List<ai.djl.ndarray.NDArray> batchBoxes, List<ai.djl.ndarray.NDArray> batchPoints, boolean selectLargest, ai.djl.ndarray.NDManager manager)
static MtcnnBatchResult	thirdStage(ai.djl.ndarray.NDManager manager, ai.djl.inference.Predictor<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList> onetPredictor, ai.djl.ndarray.NDArray imgs, ai.djl.ndarray.NDArray boxes, int w, int h, ai.djl.ndarray.NDArray scoresFiltered, ai.djl.ndarray.NDArray image_indsFiltered)

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

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

构造器详细资料

ONetModel

public ONetModel()

方法详细资料

thirdStage

public static MtcnnBatchResult thirdStage(ai.djl.ndarray.NDManager manager,
                                          ai.djl.inference.Predictor<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList> onetPredictor,
                                          ai.djl.ndarray.NDArray imgs,
                                          ai.djl.ndarray.NDArray boxes,
                                          int w,
                                          int h,
                                          ai.djl.ndarray.NDArray scoresFiltered,
                                          ai.djl.ndarray.NDArray image_indsFiltered)
                                   throws ai.djl.translate.TranslateException

抛出:

ai.djl.translate.TranslateException

processBatchBoxes

public static MtcnnBatchResult processBatchBoxes(List<ai.djl.ndarray.NDArray> batchBoxes,
                                                 List<ai.djl.ndarray.NDArray> batchPoints,
                                                 boolean selectLargest,
                                                 ai.djl.ndarray.NDManager manager)