faktel-face/io.github.cybersafetyid.faktel.face/YuNetFaceDetector

YuNetFaceDetector

class YuNetFaceDetector(engine: InferenceEngine, model: ByteArray, config: YuNetFaceDetector.Config = Config(), options: SessionOptions = SessionOptions()) : FaceDetector

Face detector backed by YuNet (face_detection_yunet_2023mar.onnx, MIT). Model I/O contract: input float32 [1,3,640,640] BGR 0..255; outputs cls|obj|bbox|kps for strides 8, 16 and 32.

Constructors

YuNetFaceDetector

constructor(engine: InferenceEngine, model: ByteArray, config: YuNetFaceDetector.Config = Config(), options: SessionOptions = SessionOptions())

Types

Config

data class Config(val scoreThreshold: Double = 0.6, val nmsThreshold: Double = 0.3, val maxFaces: Int = 10)

Functions

open override fun close()

detect

open override fun detect(image: RgbImage): List<Face>