From 29065df908aee41ab673cfc78a0d657e3063d5ef Mon Sep 17 00:00:00 2001 From: pansen <pansen626@sina.com> Date: 星期六, 12 一月 2019 15:30:49 +0800 Subject: [PATCH] Merge branch 'ywv1.2_yolodebug' --- QiaoJiaSystem/YoloServer/YoloRpcElement.h | 53 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 53 insertions(+), 0 deletions(-) diff --git a/QiaoJiaSystem/YoloServer/YoloRpcElement.h b/QiaoJiaSystem/YoloServer/YoloRpcElement.h new file mode 100644 index 0000000..bc48aeb --- /dev/null +++ b/QiaoJiaSystem/YoloServer/YoloRpcElement.h @@ -0,0 +1,53 @@ +#ifndef YOLORPCELEMENT_H +#define YOLORPCELEMENT_H + +#include <YoloServer.h> +#include <basic/pipe/PipeElement.h> +#include <basic/rpc/IceRpc.hpp> +#include <opencv2/opencv.hpp> +#include <basic/util/fastdfs/FastFds.hpp> +#include <TriggerElement.h> +//#include "../StructureApp/YoloRpcElement.h" +#include "../StructureApp/TrackingTrigger.h" + +class QSharedMemory; + +class YoloRpcElement : public basic::PipeElement { +public: + YoloRpcElement(string); + + ~YoloRpcElement(); + + void setImage(const cv::Mat &value); + + ::YoloDetect::ObjInfos getObjects() const; + + std::vector<cv::Mat> getTriggerMats(); + + std::vector<ScoredRect> getLastScoreRects() const; + + void setFdfs(FastFdsWithLock *p_fdfsClient) { + fdfsClient = p_fdfsClient; + } + + bool getTriggerState() const; + + bool getTrigger() const; + +private: + virtual void threadFunc() override; + +private: + IceRpcClient<YoloDetect::YoloDetectServerPrx> rpcClient; + cv::Mat image; + QSharedMemory *sharedMemory; + ::YoloDetect::ObjInfos objs; + TrackingTrigger *trackingTrigger; + std::vector<cv::Mat> triggerMats; + ::YoloDetect::stringData cocoData; + FastFdsWithLock *fdfsClient; + + TriggerElement m_triggerElement; +}; + +#endif // YOLORPCELEMENT_H -- Gitblit v1.8.0