From 0a5dbaaadc8b2122c853790e90311c64e5c8ef90 Mon Sep 17 00:00:00 2001 From: uttarayan21 Date: Thu, 21 Aug 2025 18:52:58 +0530 Subject: [PATCH] refactor(gui): set fixed input dimensions for face detection --- src/gui/bridge.rs | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/src/gui/bridge.rs b/src/gui/bridge.rs index f0f7c60..2142180 100644 --- a/src/gui/bridge.rs +++ b/src/gui/bridge.rs @@ -85,7 +85,6 @@ impl FaceDetectionBridge { // Load the image let img = image::open(&image_path)?; let img_rgb = img.to_rgb8(); - let (width, height) = img_rgb.dimensions(); // Convert to ndarray format let image_array = img_rgb.as_ndarray()?; @@ -94,8 +93,8 @@ impl FaceDetectionBridge { let config = FaceDetectionConfig::default() .with_threshold(threshold) .with_nms_threshold(nms_threshold) - .with_input_width(width as usize) - .with_input_height(height as usize); + .with_input_width(1024) + .with_input_height(1024); // Create detector and detect faces let faces = match executor_type { @@ -186,14 +185,14 @@ impl FaceDetectionBridge { let config1 = FaceDetectionConfig::default() .with_threshold(threshold) .with_nms_threshold(nms_threshold) - .with_input_width(img1.width() as usize) - .with_input_height(img1.height() as usize); + .with_input_width(1024) + .with_input_height(1024); let config2 = FaceDetectionConfig::default() .with_threshold(threshold) .with_nms_threshold(nms_threshold) - .with_input_width(img2.width() as usize) - .with_input_height(img2.height() as usize); + .with_input_width(1024) + .with_input_height(1024); // Create detector and embedder, detect faces and generate embeddings let (faces1, faces2, best_similarity) = match executor_type {