cn.smartjavaai.face.model.facedect

类 MtcnnFaceDetModel

java.lang.Object

cn.smartjavaai.face.model.facedect.CommonFaceDetModel

cn.smartjavaai.face.model.facedect.MtcnnFaceDetModel

所有已实现的接口:

FaceDetModel, AutoCloseable

public class MtcnnFaceDetModel
extends CommonFaceDetModel

MTCNN 人脸检测模型实现

作者:

dwj

字段概要

字段

限定符和类型	字段和说明
ai.djl.repository.zoo.ZooModel<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList>	oNetModel
ai.djl.repository.zoo.ZooModel<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList>	pNetModel
ai.djl.repository.zoo.ZooModel<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList>	rNetModel

构造器概要

构造器

构造器和说明
MtcnnFaceDetModel()

方法概要

限定符和类型	方法和说明
MtcnnPredictors	borrowPredictors()
void	close()
ai.djl.modality.cv.output.DetectedObjects	detectCore(ai.djl.modality.cv.Image image) 人脸检测
ai.djl.modality.cv.output.DetectedObjects	detectCoreByPredictors(ai.djl.modality.cv.Image image, MtcnnPredictors predictors) 使用MtcnnPredictors进行人脸检测
ai.djl.repository.zoo.ZooModel<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList>	getModel(Path modelPath, ai.djl.Device device) 加载模型
org.apache.commons.pool2.impl.GenericObjectPool<ai.djl.inference.Predictor<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList>>	getOnetPredictorPool()
org.apache.commons.pool2.impl.GenericObjectPool<ai.djl.inference.Predictor<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList>>	getPnetPredictorPool()
org.apache.commons.pool2.impl.GenericObjectPool<ai.djl.inference.Predictor<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList>>	getRnetPredictorPool()
boolean	isFromFactory()
void	loadModel(FaceDetConfig config) 加载模型
void	returnPredictor(ai.djl.inference.Predictor<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList> pNetPredictor, ai.djl.inference.Predictor<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList> rNetPredictor, ai.djl.inference.Predictor<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList> oNetPredictor)
void	setFromFactory(boolean fromFactory)

从类继承的方法 cn.smartjavaai.face.model.facedect.CommonFaceDetModel

borrowPredictor, detect, detect, detect, detect, detect, detectAndDraw, detectAndDraw, detectAndDraw, detectBase64, getPool, returnPredictor

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

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

字段详细资料

pNetModel

public ai.djl.repository.zoo.ZooModel<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList> pNetModel

rNetModel

public ai.djl.repository.zoo.ZooModel<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList> rNetModel

oNetModel

public ai.djl.repository.zoo.ZooModel<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList> oNetModel

构造器详细资料

MtcnnFaceDetModel

public MtcnnFaceDetModel()

方法详细资料

loadModel

public void loadModel(FaceDetConfig config)

加载模型

指定者:

loadModel 在接口中 FaceDetModel

覆盖:

loadModel 在类中 CommonFaceDetModel

参数:

config -

getModel

public ai.djl.repository.zoo.ZooModel<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList> getModel(Path modelPath,
                                                                                            ai.djl.Device device)
                                                                                     throws ai.djl.repository.zoo.ModelNotFoundException,
                                                                                            ai.djl.MalformedModelException,
                                                                                            IOException

加载模型

参数:

modelPath -

抛出:

ai.djl.repository.zoo.ModelNotFoundException

ai.djl.MalformedModelException

IOException

detectCore

public ai.djl.modality.cv.output.DetectedObjects detectCore(ai.djl.modality.cv.Image image)

人脸检测

指定者:

detectCore 在接口中 FaceDetModel

覆盖:

detectCore 在类中 CommonFaceDetModel

参数:

image -

返回:

detectCoreByPredictors

public ai.djl.modality.cv.output.DetectedObjects detectCoreByPredictors(ai.djl.modality.cv.Image image,
                                                                        MtcnnPredictors predictors)

使用MtcnnPredictors进行人脸检测

参数:

image -

predictors -

返回:

getPnetPredictorPool

public org.apache.commons.pool2.impl.GenericObjectPool<ai.djl.inference.Predictor<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList>> getPnetPredictorPool()

getRnetPredictorPool

public org.apache.commons.pool2.impl.GenericObjectPool<ai.djl.inference.Predictor<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList>> getRnetPredictorPool()

getOnetPredictorPool

public org.apache.commons.pool2.impl.GenericObjectPool<ai.djl.inference.Predictor<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList>> getOnetPredictorPool()

setFromFactory

public void setFromFactory(boolean fromFactory)

指定者:

setFromFactory 在接口中 FaceDetModel

覆盖:

setFromFactory 在类中 CommonFaceDetModel

isFromFactory

public boolean isFromFactory()

覆盖:

isFromFactory 在类中 CommonFaceDetModel

borrowPredictors

public MtcnnPredictors borrowPredictors()
                                 throws Exception

抛出:

Exception

returnPredictor

public void returnPredictor(ai.djl.inference.Predictor<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList> pNetPredictor,
                            ai.djl.inference.Predictor<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList> rNetPredictor,
                            ai.djl.inference.Predictor<ai.djl.ndarray.NDList,ai.djl.ndarray.NDList> oNetPredictor)

close

public void close()

指定者:

close 在接口中 AutoCloseable

覆盖:

close 在类中 CommonFaceDetModel