Compare commits
2 Commits
8467eeef4c
...
3aaac952ad
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
3aaac952ad | ||
|
|
57f5d58e1e |
@ -307,7 +307,7 @@ GstPadProbeReturn NvInferServerManager::pgie_pad_buffer_probe(
|
|||||||
uint detected_persons = 0;
|
uint detected_persons = 0;
|
||||||
float *data = static_cast<float *>(layer.buffer);
|
float *data = static_cast<float *>(layer.buffer);
|
||||||
for (unsigned int jkl = 0; jkl < 100;
|
for (unsigned int jkl = 0; jkl < 100;
|
||||||
jkl += 4) { // 100 persons for each frame
|
jkl ++) { // 100 persons for each frame
|
||||||
if (data[jkl * 57 + 4] > threshold_body_detection) {
|
if (data[jkl * 57 + 4] > threshold_body_detection) {
|
||||||
detected_persons++;
|
detected_persons++;
|
||||||
// std::cout
|
// std::cout
|
||||||
|
|||||||
@ -47,9 +47,10 @@ bool NvTrackerManager::create_nv_tracker() {
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
void NvTrackerManager::get_face_score(float *user_meta_data) {
|
float NvTrackerManager::get_face_score(float *user_meta_data) {
|
||||||
face_score =
|
face_score =
|
||||||
(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,
|
||||||
@ -228,7 +229,6 @@ GstPadProbeReturn NvTrackerManager::tracker_src_pad_buffer_probe(
|
|||||||
nvds_acquire_obj_meta_from_pool(batch_meta);
|
nvds_acquire_obj_meta_from_pool(batch_meta);
|
||||||
face_obj->unique_component_id =
|
face_obj->unique_component_id =
|
||||||
FACE_COMPONENT_ID; // Use a new 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 = *face_rect_params;
|
||||||
face_obj->rect_params.has_bg_color = 0;
|
face_obj->rect_params.has_bg_color = 0;
|
||||||
@ -236,8 +236,6 @@ GstPadProbeReturn NvTrackerManager::tracker_src_pad_buffer_probe(
|
|||||||
face_obj->rect_params.border_color =
|
face_obj->rect_params.border_color =
|
||||||
NvOSD_ColorParams{1.0, 0.0, 0.0, 1.0}; // Red box
|
NvOSD_ColorParams{1.0, 0.0, 0.0, 1.0}; // Red box
|
||||||
|
|
||||||
manager->get_face_score(user_meta_data);
|
|
||||||
|
|
||||||
// FaceCandidate *face_candidate = new FaceCandidate();
|
// FaceCandidate *face_candidate = new FaceCandidate();
|
||||||
FaceCandidTrace::FaceCandidate *face_candidate =
|
FaceCandidTrace::FaceCandidate *face_candidate =
|
||||||
new FaceCandidTrace::FaceCandidate();
|
new FaceCandidTrace::FaceCandidate();
|
||||||
@ -248,7 +246,9 @@ GstPadProbeReturn NvTrackerManager::tracker_src_pad_buffer_probe(
|
|||||||
face_candidate->frame_number = frame_meta->frame_num;
|
face_candidate->frame_number = frame_meta->frame_num;
|
||||||
face_candidate->h = face_rect_params->height;
|
face_candidate->h = face_rect_params->height;
|
||||||
face_candidate->w = face_rect_params->width;
|
face_candidate->w = face_rect_params->width;
|
||||||
face_candidate->face_score = manager->face_score;
|
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->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);
|
||||||
|
|||||||
@ -51,5 +51,5 @@ class NvTrackerManager {
|
|||||||
GstPadProbeInfo *,
|
GstPadProbeInfo *,
|
||||||
gpointer);
|
gpointer);
|
||||||
static bool check_existence(int, int, float, bool *);
|
static bool check_existence(int, int, float, bool *);
|
||||||
void get_face_score(float *);
|
float get_face_score(float *);
|
||||||
};
|
};
|
||||||
Loading…
x
Reference in New Issue
Block a user