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

Fix attached meta data

Browse Source
This commit is contained in:
Barzan Hayati 2025-07-26 14:44:47 +00:00
parent 58775bd6e3
commit 92062a514e
1 changed files with 25 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
src/nv_infer_server_manager.cpp
View File

@ -269,7 +269,7 @@ GstPadProbeReturn NvInferServerManager::pgie_pad_buffer_probe(
(void)outputBuffer; (void)outputBuffer;
// NvDsInferDims dims = outputLayersInfo[0].inferDims; // NvDsInferDims dims = outputLayersInfo[0].inferDims;
for (size_t jkl = 0; jkl < outputLayersInfo.size(); ++jkl) { for (size_t jkl = 0; jkl < outputLayersInfo.size(); jkl++) {
const NvDsInferLayerInfo &layer = outputLayersInfo[jkl]; const NvDsInferLayerInfo &layer = outputLayersInfo[jkl];
unsigned int numDims = layer.inferDims.numDims; unsigned int numDims = layer.inferDims.numDims;
@ -295,15 +295,15 @@ 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; ++jkl) { for (unsigned int jkl = 0; jkl < 100; jkl+=4) { // 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 << "x = " << data[jkl * 57 + 0] std::cout << "first for x = " << data[jkl * 57 + 0]
// << " y = " << data[jkl * 57 + 1] << " y = " << data[jkl * 57 + 1]
// << " w = " << data[jkl * 57 + 2] << " w = " << data[jkl * 57 + 2]
// << " h = " << data[jkl * 57 + 3] << " h = " << data[jkl * 57 + 3]
// << " score = " << data[jkl * 57 + 4] << << " score = " << data[jkl * 57 + 4] <<
// std::endl; std::endl;
for (unsigned int mno = 0; mno < 5; ++mno) { for (unsigned int mno = 0; mno < 5; ++mno) {
float value = data[jkl * 57 + mno]; float value = data[jkl * 57 + mno];
(void)value; (void)value;
@ -324,15 +324,24 @@ GstPadProbeReturn NvInferServerManager::pgie_pad_buffer_probe(
NvOSD_TextParams &text_params = obj_meta->text_params; NvOSD_TextParams &text_params = obj_meta->text_params;
/* Assign bounding box coordinates. */ /* Assign bounding box coordinates. */
rect_params.left = rect_params.left =
data[index * 57 + 0] * MUXER_OUTPUT_WIDTH / PGIE_NET_WIDTH; int(data[index * 57 + 0] * MUXER_OUTPUT_WIDTH / PGIE_NET_WIDTH);
rect_params.top = data[index * 57 + 1] * MUXER_OUTPUT_HEIGHT / rect_params.top = int(data[index * 57 + 1] * MUXER_OUTPUT_HEIGHT /
PGIE_NET_HEIGHT; PGIE_NET_HEIGHT);
rect_params.width = rect_params.width =
(data[index * 57 + 2] - data[index * 57 + 0]) * int((data[index * 57 + 2] - data[index * 57 + 0]) *
MUXER_OUTPUT_WIDTH / PGIE_NET_WIDTH; MUXER_OUTPUT_WIDTH / PGIE_NET_WIDTH);
rect_params.height = rect_params.height =
(data[index * 57 + 3] - data[index * 57 + 1]) * int((data[index * 57 + 3] - data[index * 57 + 1]) *
MUXER_OUTPUT_HEIGHT / PGIE_NET_HEIGHT; MUXER_OUTPUT_HEIGHT / PGIE_NET_HEIGHT);
std::cout << "second for x = " << rect_params.left
<< " y = " << rect_params.top
<< " w = " << rect_params.top
<< " h = " << rect_params.width
<< " score = " << obj_meta->confidence <<
std::endl;
/* Border of width 3. */ /* Border of width 3. */
rect_params.border_width = 3; rect_params.border_width = 3;
rect_params.has_bg_color = 0; rect_params.has_bg_color = 0;
@ -364,6 +373,6 @@ GstPadProbeReturn NvInferServerManager::pgie_pad_buffer_probe(
} }
} }
} }
use_device_mem = 1 - use_device_mem; // use_device_mem = 1 - use_device_mem;
return GST_PAD_PROBE_OK; return GST_PAD_PROBE_OK;
} }