diff --git a/src/nv_infer_server_manager.cpp b/src/nv_infer_server_manager.cpp index 8b542d6..2314d22 100644 --- a/src/nv_infer_server_manager.cpp +++ b/src/nv_infer_server_manager.cpp @@ -7,6 +7,7 @@ #define MAX_DISPLAY_LEN 64 #define PGIE_CLASS_ID_PERSON 0 +#define CLASS_ID_IMPRECISE_FACE 1 #define PGIE_DETECTED_CLASS_NUM 1 gint NvInferServerManager::frame_number = 0; @@ -350,7 +351,7 @@ void NvInferServerManager::update_frame_with_face_body_meta( imprecise_face_obj_meta ->obj_label[sizeof(imprecise_face_obj_meta->obj_label) - 1] = '\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 // 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->object_id = UNTRACKED_OBJECT_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 = imprecise_face_obj_meta->rect_params; NvOSD_TextParams &text_params_imprecise_face =