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

Add object label in nvinfer

Browse Source
This commit is contained in:
Barzan Hayati 2025-08-24 22:33:11 +00:00
parent a6a1a5196c
commit c91e5af0d0
2 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/nv_infer_server_manager.cpp
View File

@ -345,6 +345,11 @@ void NvInferServerManager::update_frame_with_face_body_meta(
// the config for the inference component that produced these // the config for the inference component that produced these
// tensor outputs. // tensor outputs.
strncpy(imprecise_face_obj_meta->obj_label, imprecise_face_str[0],
sizeof(imprecise_face_obj_meta->obj_label) - 1);
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;
// imprecise_face_obj_meta->unique_component_id // imprecise_face_obj_meta->unique_component_id
@ -416,6 +421,10 @@ void NvInferServerManager::update_frame_with_face_body_meta(
NvDsObjectMeta *body_obj_meta = NvDsObjectMeta *body_obj_meta =
nvds_acquire_obj_meta_from_pool(batch_meta); nvds_acquire_obj_meta_from_pool(batch_meta);
strncpy(body_obj_meta->obj_label, pgie_class_str[0],
sizeof(body_obj_meta->obj_label) - 1);
body_obj_meta->obj_label[sizeof(body_obj_meta->obj_label) - 1] =
'\0'; // Ensure null-termination
body_obj_meta->unique_component_id = meta->unique_id; body_obj_meta->unique_component_id = meta->unique_id;
body_obj_meta->confidence = data[index * 57 + 4]; body_obj_meta->confidence = data[index * 57 + 4];
// body_obj_meta->object_id = UNTRACKED_OBJECT_ID; // body_obj_meta->object_id = UNTRACKED_OBJECT_ID;
@ -468,8 +477,6 @@ void NvInferServerManager::update_frame_with_face_body_meta(
NvInferServerManager::Point2D NvInferServerManager::Point2D
NvInferServerManager::find_left_down_corner_shoulder(float *data, uint index) { NvInferServerManager::find_left_down_corner_shoulder(float *data, uint index) {
Point2D left_down_shoulder; Point2D left_down_shoulder;
// float x_shoulder;
// float y_shoulder;
if (data[index * 57 + 21] > data[index * 57 + 24]) { if (data[index * 57 + 21] > data[index * 57 + 24]) {
left_down_shoulder.x = data[index * 57 + 21]; left_down_shoulder.x = data[index * 57 + 21];
left_down_shoulder.y = data[index * 57 + 22]; left_down_shoulder.y = data[index * 57 + 22];

7
src/nv_tracker_manager.cpp
View File

@ -48,9 +48,8 @@ bool NvTrackerManager::create_nv_tracker() {
} }
float NvTrackerManager::get_face_score(float *user_meta_data) { float NvTrackerManager::get_face_score(float *user_meta_data) {
face_score = return (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,
@ -248,7 +247,7 @@ GstPadProbeReturn NvTrackerManager::tracker_src_pad_buffer_probe(
face_candidate->w = face_rect_params->width; face_candidate->w = face_rect_params->width;
face_candidate->face_score = manager->get_face_score( face_candidate->face_score = manager->get_face_score(
user_meta_data); // manager->face_score; user_meta_data); // manager->face_score;
face_obj->confidence = face_candidate->face_score; //1.0; 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);