| | |
| | | #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; |
| | |
| | | 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 |