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

Modify imprecise face class id

Browse Source
This commit is contained in:
Barzan Hayati 2025-08-24 22:45:13 +00:00
parent c91e5af0d0
commit 54e775f304
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/nv_infer_server_manager.cpp
View File

@ -7,6 +7,7 @@
#define MAX_DISPLAY_LEN 64 #define MAX_DISPLAY_LEN 64
#define PGIE_CLASS_ID_PERSON 0 #define PGIE_CLASS_ID_PERSON 0
#define CLASS_ID_IMPRECISE_FACE 1
#define PGIE_DETECTED_CLASS_NUM 1 #define PGIE_DETECTED_CLASS_NUM 1
gint NvInferServerManager::frame_number = 0; gint NvInferServerManager::frame_number = 0;
@ -350,7 +351,7 @@ void NvInferServerManager::update_frame_with_face_body_meta(
imprecise_face_obj_meta imprecise_face_obj_meta
->obj_label[sizeof(imprecise_face_obj_meta->obj_label) - 1] = ->obj_label[sizeof(imprecise_face_obj_meta->obj_label) - 1] =
'\0'; // Ensure null-termination '\0'; // Ensure null-termination
imprecise_face_obj_meta->unique_component_id = meta->unique_id; imprecise_face_obj_meta->unique_component_id = meta->unique_id; // 1
// imprecise_face_obj_meta->unique_component_id // imprecise_face_obj_meta->unique_component_id
// Meaning: The ID of the component (PGIE, SGIE, Tracker, // Meaning: The ID of the component (PGIE, SGIE, Tracker,
@ -369,7 +370,7 @@ void NvInferServerManager::update_frame_with_face_body_meta(
imprecise_face_obj_meta->confidence = data[index * 57 + 4]; imprecise_face_obj_meta->confidence = data[index * 57 + 4];
// imprecise_face_obj_meta->object_id = UNTRACKED_OBJECT_ID; // imprecise_face_obj_meta->object_id = UNTRACKED_OBJECT_ID;
imprecise_face_obj_meta->class_id = imprecise_face_obj_meta->class_id =
PGIE_CLASS_ID_PERSON; // 0 for body detection CLASS_ID_IMPRECISE_FACE; // 0 for body detection
NvOSD_RectParams &rect_params_imprecise_face = NvOSD_RectParams &rect_params_imprecise_face =
imprecise_face_obj_meta->rect_params; imprecise_face_obj_meta->rect_params;
NvOSD_TextParams &text_params_imprecise_face = NvOSD_TextParams &text_params_imprecise_face =