From abd9036f024a77d22b47a108bc88b27b9732e3c5 Mon Sep 17 00:00:00 2001 From: zhangjixing <775834166@qq.com> Date: 星期二, 15 一月 2019 17:27:27 +0800 Subject: [PATCH] --- QiaoJiaSystem/StructureApp/YoloRpcElement.h | 23 ++++++----------------- 1 files changed, 6 insertions(+), 17 deletions(-) diff --git a/QiaoJiaSystem/StructureApp/YoloRpcElement.h b/QiaoJiaSystem/StructureApp/YoloRpcElement.h index 857f2d7..8434ba6 100644 --- a/QiaoJiaSystem/StructureApp/YoloRpcElement.h +++ b/QiaoJiaSystem/StructureApp/YoloRpcElement.h @@ -8,31 +8,22 @@ #include <basic/util/fastdfs/FastFds.hpp> #include <RapidStructureApp/TriggerElement.h> #include "TrackingTrigger.h" - +#include "DBStruct.h" class QSharedMemory; class YoloRpcElement : public basic::PipeElement { public: - YoloRpcElement(string); + YoloRpcElement(std::string shareMemoryName); ~YoloRpcElement(); void setImage(const cv::Mat &value); + cv::Mat getImage() const; + ::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; @@ -41,12 +32,10 @@ cv::Mat image; QSharedMemory *sharedMemory; ::YoloDetect::ObjInfos objs; - TrackingTrigger *trackingTrigger; - std::vector<cv::Mat> triggerMats; ::YoloDetect::stringData cocoData; - FastFdsWithLock *fdfsClient; + std::vector<ScoredRect> triggerScoredRects; + TrackingTrigger* trackingTrigger; - TriggerElement m_triggerElement; }; #endif // YOLORPCELEMENT_H -- Gitblit v1.8.0