feat: Added stuff

src/main.rs

@@ -4,10 +4,12 @@ use bounding_box::roi::MultiRoi;
 use detector::{facedet::retinaface::FaceDetectionConfig, faceembed};
 use errors::*;
 use fast_image_resize::ResizeOptions;
-use nalgebra::zero;
+use ndarray::*;
 use ndarray_image::*;
+use ndarray_resize::NdFir;
 const RETINAFACE_MODEL: &[u8] = include_bytes!("../models/retinaface.mnn");
 const FACENET_MODEL: &[u8] = include_bytes!("../models/facenet.mnn");
+const CHUNK_SIZE: usize = 8;
 pub fn main() -> Result<()> {
     tracing_subscriber::fmt()
         .with_env_filter("trace")
@@ -19,10 +21,16 @@ pub fn main() -> Result<()> {
     match args.cmd {
         cli::SubCommand::Detect(detect) => {
             use detector::facedet;
-            let retinaface = facedet::retinaface::FaceDetection::new_from_bytes(RETINAFACE_MODEL)
+            let retinaface = facedet::retinaface::FaceDetection::builder()(RETINAFACE_MODEL)
+                .change_context(Error)?
+                .with_forward_type(detect.forward_type)
+                .build()
                 .change_context(errors::Error)
                 .attach_printable("Failed to create face detection model")?;
-            let facenet = faceembed::facenet::EmbeddingGenerator::new_from_bytes(FACENET_MODEL)
+            let facenet = faceembed::facenet::EmbeddingGenerator::builder()(FACENET_MODEL)
+                .change_context(Error)?
+                .with_forward_type(detect.forward_type)
+                .build()
                 .change_context(errors::Error)
                 .attach_printable("Failed to create face embedding model")?;
             let image = image::open(detect.image).change_context(Error)?;
@@ -45,8 +53,6 @@ pub fn main() -> Result<()> {
                 use bounding_box::draw::*;
                 array.draw(bbox, color::palette::css::GREEN_YELLOW.to_rgba8(), 1);
             }
-            use ndarray::{Array2, Array3, Array4, Axis};
-            use ndarray_resize::NdFir;
             let face_rois = array
                 .view()
                 .multi_roi(&output.bbox)
@@ -68,21 +74,19 @@ pub fn main() -> Result<()> {
                 .collect::<Result<Vec<_>>>()?;
             let face_roi_views = face_rois.iter().map(|roi| roi.view()).collect::<Vec<_>>();
 
+            let chunk_size = CHUNK_SIZE;
             let embeddings = face_roi_views
-                .chunks(8)
+                .chunks(chunk_size)
                 .map(|chunk| {
                     tracing::info!("Processing chunk of size: {}", chunk.len());
 
                     if chunk.len() < 8 {
                         tracing::warn!("Chunk size is less than 8, padding with zeros");
                         let zeros = Array3::zeros((512, 512, 3));
-                        let padded: Vec<ndarray::ArrayView3<'_, u8>> = chunk
-                            .iter()
-                            .cloned()
-                            .chain(core::iter::repeat(zeros.view()))
-                            .take(8)
-                            .collect();
-                        let face_rois: Array4<u8> = ndarray::stack(Axis(0), padded.as_slice())
+                        let zero_array = core::iter::repeat(zeros.view())
+                            .take(chunk_size)
+                            .collect::<Vec<_>>();
+                        let face_rois: Array4<u8> = ndarray::stack(Axis(0), zero_array.as_slice())
                             .change_context(errors::Error)
                             .attach_printable("Failed to stack rois together")?;
                         let output = facenet.run_models(face_rois.view()).change_context(Error)?;