haiati/FaceRecognition
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix face score

Browse Source
This commit is contained in:
Barzan Hayati 2025-08-13 07:33:19 +00:00
parent 57f5d58e1e
commit 3aaac952ad
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/nv_tracker_manager.cpp
View File

@ -47,9 +47,10 @@ bool NvTrackerManager::create_nv_tracker() {
return true; return true;
} }
void NvTrackerManager::get_face_score(float *user_meta_data) { float NvTrackerManager::get_face_score(float *user_meta_data) {
face_score = face_score =
(user_meta_data[8] + user_meta_data[11] + user_meta_data[14]) / 3; (user_meta_data[8] + user_meta_data[11] + user_meta_data[14]) / 3;
return face_score;
} }
bool NvTrackerManager::check_existence(int object_id, int source_id, float area, bool NvTrackerManager::check_existence(int object_id, int source_id, float area,
@ -228,7 +229,6 @@ GstPadProbeReturn NvTrackerManager::tracker_src_pad_buffer_probe(
nvds_acquire_obj_meta_from_pool(batch_meta); nvds_acquire_obj_meta_from_pool(batch_meta);
face_obj->unique_component_id = face_obj->unique_component_id =
FACE_COMPONENT_ID; // Use a new component ID FACE_COMPONENT_ID; // Use a new component ID
face_obj->confidence = 1.0;
face_obj->rect_params = *face_rect_params; face_obj->rect_params = *face_rect_params;
face_obj->rect_params.has_bg_color = 0; face_obj->rect_params.has_bg_color = 0;
@ -236,8 +236,6 @@ GstPadProbeReturn NvTrackerManager::tracker_src_pad_buffer_probe(
face_obj->rect_params.border_color = face_obj->rect_params.border_color =
NvOSD_ColorParams{1.0, 0.0, 0.0, 1.0}; // Red box NvOSD_ColorParams{1.0, 0.0, 0.0, 1.0}; // Red box
manager->get_face_score(user_meta_data);
// FaceCandidate *face_candidate = new FaceCandidate(); // FaceCandidate *face_candidate = new FaceCandidate();
FaceCandidTrace::FaceCandidate *face_candidate = FaceCandidTrace::FaceCandidate *face_candidate =
new FaceCandidTrace::FaceCandidate(); new FaceCandidTrace::FaceCandidate();
@ -248,7 +246,9 @@ GstPadProbeReturn NvTrackerManager::tracker_src_pad_buffer_probe(
face_candidate->frame_number = frame_meta->frame_num; face_candidate->frame_number = frame_meta->frame_num;
face_candidate->h = face_rect_params->height; face_candidate->h = face_rect_params->height;
face_candidate->w = face_rect_params->width; face_candidate->w = face_rect_params->width;
face_candidate->face_score = manager->face_score; face_candidate->face_score = manager->get_face_score(
user_meta_data); // manager->face_score;
face_obj->confidence = face_candidate->face_score; //1.0;
face_candidate->object_id = obj_meta->object_id; face_candidate->object_id = obj_meta->object_id;
face_candidate->source_id = frame_meta->source_id; face_candidate->source_id = frame_meta->source_id;
bool add_status = face_candidate_trace->add(face_candidate); bool add_status = face_candidate_trace->add(face_candidate);

2
src/nv_tracker_manager.hpp
View File

@ -51,5 +51,5 @@ class NvTrackerManager {
GstPadProbeInfo *, GstPadProbeInfo *,
gpointer); gpointer);
static bool check_existence(int, int, float, bool *); static bool check_existence(int, int, float, bool *);
void get_face_score(float *); float get_face_score(float *);
}; };