diff --git a/src/nv_infer_server_manager.cpp b/src/nv_infer_server_manager.cpp index b864355..8b542d6 100644 --- a/src/nv_infer_server_manager.cpp +++ b/src/nv_infer_server_manager.cpp @@ -345,6 +345,11 @@ void NvInferServerManager::update_frame_with_face_body_meta( // the config for the inference component that produced these // 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 @@ -416,6 +421,10 @@ void NvInferServerManager::update_frame_with_face_body_meta( NvDsObjectMeta *body_obj_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->confidence = data[index * 57 + 4]; // body_obj_meta->object_id = UNTRACKED_OBJECT_ID; @@ -468,8 +477,6 @@ void NvInferServerManager::update_frame_with_face_body_meta( NvInferServerManager::Point2D NvInferServerManager::find_left_down_corner_shoulder(float *data, uint index) { Point2D left_down_shoulder; - // float x_shoulder; - // float y_shoulder; if (data[index * 57 + 21] > data[index * 57 + 24]) { left_down_shoulder.x = data[index * 57 + 21]; left_down_shoulder.y = data[index * 57 + 22]; diff --git a/src/nv_tracker_manager.cpp b/src/nv_tracker_manager.cpp index aae7032..79f4fea 100644 --- a/src/nv_tracker_manager.cpp +++ b/src/nv_tracker_manager.cpp @@ -48,9 +48,8 @@ bool NvTrackerManager::create_nv_tracker() { } float NvTrackerManager::get_face_score(float *user_meta_data) { - face_score = - (user_meta_data[8] + user_meta_data[11] + user_meta_data[14]) / 3; - return face_score; + return (user_meta_data[8] + user_meta_data[11] + user_meta_data[14]) / 3; + ; } 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->face_score = manager->get_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->source_id = frame_meta->source_id; bool add_status = face_candidate_trace->add(face_candidate);