From 97b99ec6ed9e9ee1e83e0ec301339c74d67e3dc5 Mon Sep 17 00:00:00 2001
From: Barzan Hayati <barzanhayati@gmail.com>
Date: Wed, 10 Sep 2025 22:31:58 +0000
Subject: [PATCH] Optimize check embedding zero

---
 src/face_nv_infer_server_manager.cpp | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/src/face_nv_infer_server_manager.cpp b/src/face_nv_infer_server_manager.cpp
index 83f446c..52e1af2 100644
--- a/src/face_nv_infer_server_manager.cpp
+++ b/src/face_nv_infer_server_manager.cpp
@@ -1386,7 +1386,20 @@ GstPadProbeReturn FaceNvInferServerManager::sgie_pad_buffer_probe(
                             // }
                         }
                     } else {
-                        is_zero_embedding_vector = allZero(data_face, 512);
+                        // is_zero_embedding_vector = 0: send data into server!
+                        if (data_face[0] ==
+                            0.0f)  // if the first element is zero, then its
+                                   // possible all element are zero
+                        {
+                            is_zero_embedding_vector = allZero(data_face, 512);
+                            // is_zero_embedding_vector = 1 where approximately
+                            // 3 times more than is_zero_embedding_vector 0
+                        } else {
+                            is_zero_embedding_vector =
+                                std::all_of(data_face, data_face + 512,
+                                            [](float v) { return v == 0.0f; });
+                            // is_zero_embedding_vector = 0;
+                        }
                         // std::cout<<"is_zero_embedding_vector =
                         // "<<is_zero_embedding_vector<<std::endl; for (int l =
                         // 0; l < 512; l++) {