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