FaceDetectionModelBase

abstract class FaceDetectionModelBase<I>(modelKindDescription: String?) : OnnxHighLevelModel<I, List<<ERROR CLASS>>>

Base class for face detection models.

Constructors

common

fun FaceDetectionModelBase(modelKindDescription: String? = null)

Types

Companion

common

object Companion

Functions

convert

common

open override fun convert(output: <ERROR CLASS>): List<<ERROR CLASS>>

Converts raw model output to the result.

detectFaces

common

fun detectFaces(image: I, topK: Int = 5, iouThreshold: Float = 0.5f): List<<ERROR CLASS>>

Detects topK faces on the given image. If topK is negative all detected faces are returned.

initializeWith

common

open override fun initializeWith(vararg executionProviders: ExecutionProvider)

Initialize the model with the specified executions providers.

predict

common

open fun predict(input: I): List<<ERROR CLASS>>

Makes prediction on the given input.

summary

common

open fun summary(): <ERROR CLASS>

Properties

internalModel

common

abstract val internalModel: OnnxInferenceModel

Model used to make predictions.

modelKindDescription

common

open override val modelKindDescription: String? = null

High-level description of the model. Used for model summary printing. For the model hub models it equals to the string representation of the OnnxModelType

preprocessing

common

abstract val preprocessing: <ERROR CLASS><I, <ERROR CLASS><FloatArray, <ERROR CLASS>>>

Preprocessing operation specific to this model.

Inheritors

FaceDetectionModel

Extensions

detectFaces

android

fun FaceDetectionModelBase<Bitmap>.detectFaces(imageProxy: ImageProxy, topK: Int = 5, iouThreshold: Float = 0.5f): List<DetectedObject>

Detects topK faces on the given imageProxy. If topK is negative all detected faces are returned.