Package org.jetbrains.kotlinx.dl.onnx.inference.facealignment

Types

FaceAlignmentModelBase
Link copied to clipboard
common
abstract class FaceAlignmentModelBase<I>(modelKindDescription: String?) : OnnxHighLevelModel<I, List<<ERROR CLASS>>>

Base class for face alignment models.

FaceDetectionModel
Link copied to clipboard
class FaceDetectionModel(internalModel: OnnxInferenceModel, modelKindDescription: String?) : FaceDetectionModelBase<Bitmap> , CameraXCompatibleModel, InferenceModel
class FaceDetectionModel(internalModel: OnnxInferenceModel, modelKindDescription: String?) : FaceDetectionModelBase<BufferedImage> , InferenceModel

Face detection model implementation.

FaceDetectionModelBase
Link copied to clipboard
common
abstract class FaceDetectionModelBase<I>(modelKindDescription: String?) : OnnxHighLevelModel<I, List<<ERROR CLASS>>>

Base class for face detection models.

Fan2D106FaceAlignmentModel
Link copied to clipboard
class Fan2D106FaceAlignmentModel(internalModel: OnnxInferenceModel, modelKindDescription: String?) : FaceAlignmentModelBase<Bitmap> , CameraXCompatibleModel, InferenceModel

The light-weight API for solving Face Alignment task.

class Fan2D106FaceAlignmentModel(internalModel: OnnxInferenceModel, modelKindDescription: String?) : FaceAlignmentModelBase<BufferedImage> , InferenceModel

The light-weight API for solving Face Alignment task via Fan2D106 model.

Functions

detectFaces
Link copied to clipboard
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.

detectLandmarks
Link copied to clipboard
fun FaceAlignmentModelBase<Bitmap>.detectLandmarks(imageProxy: ImageProxy): List<Landmark>

Detects Landmark objects on the given imageProxy.