Skip to main content

Module ml

Module ml 

Source
Expand description

Machine Learning module

Provides face detection and recognition using ONNX models.

Re-exports§

pub use clustering::ClusterInput;
pub use clustering::FaceClusterer;
pub use face_detector::FaceDetector;
pub use face_embedder::EmbedderConfig;
pub use face_embedder::FaceEmbedder;
pub use face_embedder::FaceEmbedding;
pub use face_embedder::LocalEmbedder;
pub use resolver::rerank;
pub use resolver::ResolverContext;
pub use resolver::ResolverWeights;
pub use retrieval::retrieve_candidates;
pub use retrieval::BandingConfig;
pub use retrieval::ConfidenceBand;
pub use retrieval::RetrievalHit;
pub use runtime::active_execution_provider;
pub use runtime::OnnxRuntime;

Modules§

alignment
Face alignment via 5-point similarity transform.
clustering
Face clustering using agglomerative complete-linkage.
face_detector
SCRFD Face Detection
face_embedder
ArcFace Face Embedding
remote_embedder
Remote GPU bridge embedder.
resolver
Context-aware resolver for face retrieval.
retrieval
Gallery-based face retrieval.
runtime
ONNX Runtime initialization and management