Skip to main content

smriti/ml/
mod.rs

1//! Machine Learning module
2//!
3//! Provides face detection and recognition using ONNX models.
4
5pub mod alignment;
6pub mod clustering;
7pub mod face_detector;
8pub mod face_embedder;
9pub mod remote_embedder;
10pub mod resolver;
11pub mod retrieval;
12pub mod runtime;
13
14pub use clustering::{ClusterInput, FaceClusterer};
15pub use face_detector::FaceDetector;
16pub use face_embedder::{EmbedderConfig, FaceEmbedder, FaceEmbedding, LocalEmbedder};
17pub use resolver::{rerank, ResolverContext, ResolverWeights};
18pub use retrieval::{retrieve_candidates, BandingConfig, ConfidenceBand, RetrievalHit};
19pub use runtime::{active_execution_provider, OnnxRuntime};