From d3b7bbe7102cd089680a828f5d8f6402c8cf6342 Mon Sep 17 00:00:00 2001 From: pansen <pansen626@sina.com> Date: 星期四, 07 三月 2019 14:43:28 +0800 Subject: [PATCH] GB28181集成完成,集成推流模块 --- QiaoJiaSystem/YoloServer/ImageDrawElement.h | 30 ++++++++++++++++++++---------- 1 files changed, 20 insertions(+), 10 deletions(-) diff --git a/QiaoJiaSystem/YoloServer/ImageDrawElement.h b/QiaoJiaSystem/YoloServer/ImageDrawElement.h index 03fd9df..a9c2a51 100644 --- a/QiaoJiaSystem/YoloServer/ImageDrawElement.h +++ b/QiaoJiaSystem/YoloServer/ImageDrawElement.h @@ -1,20 +1,30 @@ #ifndef IMAGEDRAWELEMENT_H #define IMAGEDRAWELEMENT_H -#include <basic/pipe_element/ImageFactoryElement.h> -#include <mutex> -class ImageDrawElement : public ImageFactoryElement -{ +#include <YoloServer.h> +//#include <FaceServer.h> +#include <basic/pipe_element/ImageFactoryElement.h> +#include <basic/core/DoubleBufferedData.h> +#include "../StructureApp/TrackingTrigger.h" + +class ImageDrawElement : public ImageFactoryElement { public: ImageDrawElement(); - void setRects(const std::vector<cv::Rect2f> &value); + + // ImageFactoryElement interface +// void setFaces(std::vector<ScoredRect> value); + + void setYoloObjects(std::vector<ScoredRect> value); private: - std::mutex rectsMtx; - std::vector<cv::Rect2f> rects; - std::vector<cv::Rect2f> rectsBuffer; - bool rectsUpdated; - virtual void processImage(cv::Mat& image); + virtual void processImage(cv::Mat &) override; + + void darwProperty(cv::Mat &image, string key, string value, int x, int y); + +private: +// DoubleBufferedData<std::vector<ScoredRect>> faces; + DoubleBufferedData<std::vector<ScoredRect>> yoloObjects; }; + #endif // IMAGEDRAWELEMENT_H -- Gitblit v1.8.0