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

Compare commits

base: haiati:3aaac952ad83992d72a243eb3a0118a92a1c2565
Branches Tags
haiati:master
haiati:refactor-draw-imprecise-face
haiati:add-face-nv-infer-server-to-pipeline
haiati:attach-objects-meta-data-to-gst-dsexample-probe
haiati:nvurisrcbin
haiati:urisourcebin
haiati:uridecodebin3
haiati:total-execution-time
..
compare: haiati:8467eeef4cb18838d52e8d570b6371f4791f9d33
Branches Tags
haiati:master
haiati:refactor-draw-imprecise-face
haiati:add-face-nv-infer-server-to-pipeline
haiati:attach-objects-meta-data-to-gst-dsexample-probe
haiati:nvurisrcbin
haiati:urisourcebin
haiati:uridecodebin3
haiati:total-execution-time

No commits in common. "3aaac952ad83992d72a243eb3a0118a92a1c2565" and "8467eeef4cb18838d52e8d570b6371f4791f9d33" have entirely different histories.
3aaac952ad ... 8467eeef4c

3 changed files with 7 additions and 7 deletions
Show Stats Expand all files Collapse all files

2
src/nv_infer_server_manager.cpp
View File

@ -307,7 +307,7 @@ GstPadProbeReturn NvInferServerManager::pgie_pad_buffer_probe(
uint detected_persons = 0;
float *data = static_cast<float *>(layer.buffer);
for (unsigned int jkl = 0; jkl < 100;
jkl ++) { // 100 persons for each frame
jkl += 4) { // 100 persons for each frame
if (data[jkl * 57 + 4] > threshold_body_detection) {
detected_persons++;
// std::cout

10
src/nv_tracker_manager.cpp
View File

@ -47,10 +47,9 @@ bool NvTrackerManager::create_nv_tracker() {
return true;
}
float NvTrackerManager::get_face_score(float *user_meta_data) {
void 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;
}
bool NvTrackerManager::check_existence(int object_id, int source_id, float area,
@ -229,6 +228,7 @@ GstPadProbeReturn NvTrackerManager::tracker_src_pad_buffer_probe(
nvds_acquire_obj_meta_from_pool(batch_meta);
face_obj->unique_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.has_bg_color = 0;
@ -236,6 +236,8 @@ GstPadProbeReturn NvTrackerManager::tracker_src_pad_buffer_probe(
face_obj->rect_params.border_color =
NvOSD_ColorParams{1.0, 0.0, 0.0, 1.0}; // Red box
manager->get_face_score(user_meta_data);
// FaceCandidate *face_candidate = new FaceCandidate();
FaceCandidTrace::FaceCandidate *face_candidate =
new FaceCandidTrace::FaceCandidate();
@ -246,9 +248,7 @@ GstPadProbeReturn NvTrackerManager::tracker_src_pad_buffer_probe(
face_candidate->frame_number = frame_meta->frame_num;
face_candidate->h = face_rect_params->height;
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_candidate->face_score = manager->face_score;
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);

2
src/nv_tracker_manager.hpp
View File

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