mirror of
https://github.com/wwhai/generic-rtsp-yolov8-render.git
synced 2026-04-22 23:27:21 +08:00
wwhai
91 lines
3.5 KiB
C++
91 lines
3.5 KiB
C++
// Copyright (C) 2025 wwhai
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as
|
|
// published by the Free Software Foundation, either version 3 of the
|
|
// License, or (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
#include "detection_thread.h"
|
|
#include <stdio.h>
|
|
#include "thread_args.h"
|
|
#include <stdlib.h>
|
|
#include <time.h>
|
|
#include <unistd.h>
|
|
#include "opencv_dnn_module.h"
|
|
#include "opencv_utils.h"
|
|
#include "warning_timer.h"
|
|
#include "timestamp_utils.h"
|
|
#include "logger.h"
|
|
void *frame_detection_thread(void *arg)
|
|
{
|
|
const ThreadArgs *args = (ThreadArgs *)arg;
|
|
const char *modelPath = "./yolov8n.onnx";
|
|
cv::dnn::Net net;
|
|
if (Init_CV_ONNX_DNN_Yolov8(modelPath, &net) != 0)
|
|
{
|
|
log_info( "Error: Failed to initialize the YOLOv8 ONNX DNN model.");
|
|
pthread_exit(NULL);
|
|
return NULL;
|
|
}
|
|
|
|
while (1)
|
|
{
|
|
if (args->ctx->is_cancelled)
|
|
{
|
|
goto END;
|
|
}
|
|
|
|
QueueItem detection_item;
|
|
memset(&detection_item, 0, sizeof(QueueItem));
|
|
if (dequeue(args->detection_queue, &detection_item))
|
|
{
|
|
if (detection_item.type == ONLY_FRAME)
|
|
{
|
|
AVFrame *detection_frame = (AVFrame *)detection_item.data;
|
|
cv::Mat detection_mat = AVFrameToCVMat(detection_frame);
|
|
if (!detection_mat.empty())
|
|
{
|
|
std::vector<Box> outputs;
|
|
Infer_CV_ONNX_DNN_Yolov8(&net, detection_mat, outputs);
|
|
QueueItem boxes_item;
|
|
memset(&boxes_item, 0, sizeof(QueueItem));
|
|
boxes_item.box_count = (outputs.size() > 20) ? 20 : outputs.size();
|
|
boxes_item.type = ONLY_BOXES;
|
|
boxes_item.data = NULL;
|
|
for (int i = 0; i < boxes_item.box_count; ++i)
|
|
{
|
|
// 存储检测框信息
|
|
boxes_item.Boxes[i].x = outputs[i].x;
|
|
boxes_item.Boxes[i].y = outputs[i].y;
|
|
boxes_item.Boxes[i].w = outputs[i].w;
|
|
boxes_item.Boxes[i].h = outputs[i].h;
|
|
boxes_item.Boxes[i].prop = outputs[i].prop;
|
|
strcpy(boxes_item.Boxes[i].label, outputs[i].label);
|
|
// 测试：检测到人以后 POST 具体的识别JSON
|
|
if (strcmp(boxes_item.Boxes[i].label, "person") == 0)
|
|
{
|
|
if (detection_frame->width > 0 && detection_frame->height > 0)
|
|
{
|
|
warning_timer_record_warning(boxes_item.Boxes[i].label, get_current_timestamp(), detection_frame);
|
|
}
|
|
}
|
|
}
|
|
enqueue(args->box_queue, boxes_item);
|
|
}
|
|
av_frame_free(&detection_frame);
|
|
}
|
|
}
|
|
}
|
|
|
|
END:
|
|
Release_CV_ONNX_DNN_Yolov8(&net);
|
|
pthread_exit(NULL);
|
|
return NULL;
|
|
} |