servius/face-detector
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: implement the facenet implementation for ort

Browse Source
This commit is contained in:
uttarayan21
2025-08-18 13:20:55 +05:30
parent 5a1f4b9ef6
commit e7c9c38ed7
4 changed files with 62 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/faceembed/facenet/mnn.rs
View File

@@ -142,14 +142,6 @@ impl EmbeddingGenerator {
.change_context(Error)?;
Ok(output)
}
// pub fn embedding(&self, roi: ArrayView3<u8>) -> Result<Array1<u8>> {
// todo!()
// }
// pub fn embeddings(&self, roi: ArrayView4<u8>) -> Result<Array2<u8>> {
// todo!()
// }
}
impl FaceNetEmbedder for EmbeddingGenerator {
@@ -160,7 +152,7 @@ impl FaceNetEmbedder for EmbeddingGenerator {
// Main trait implementation for backward compatibility
impl crate::faceembed::FaceEmbedder for EmbeddingGenerator {
fn run_models(&self, faces: ArrayView4<u8>) -> Result<Array2<f32>> {
self.run_models(faces)
fn run_models(&mut self, faces: ArrayView4<u8>) -> Result<Array2<f32>> {
EmbeddingGenerator::run_models(self, faces)
}
}