#ifndef PERSONELEMENT_H
|
#define PERSONELEMENT_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 <RapidStructureApp/TriggerElement.h>
|
#include "TrackingTrigger.h"
|
#include "DBStruct.h"
|
class QSharedMemory;
|
|
class PersonElement : public basic::PipeElement {
|
public:
|
PersonElement(){}
|
PersonElement(const SdkRule& sdkRule);
|
|
~PersonElement();
|
|
void setImage(const cv::Mat &value);
|
|
std::vector<cv::Mat> getTriggerMats();
|
|
std::vector<ScoredRect> getLastScoreRects() const;
|
|
void setFdfs(FastFdsWithLock *p_fdfsClient) {
|
fdfsClient = p_fdfsClient;
|
}
|
|
bool getTriggerState() const;
|
|
bool getTrigger() const;
|
void setObjsResults(const std::vector<ScoredRect>& objs);
|
|
private:
|
virtual void threadFunc() override;
|
|
private:
|
// IceRpcClient<YoloDetect::YoloDetectServerPrx> rpcClient;
|
cv::Mat image;
|
// QSharedMemory *sharedMemory;
|
// ::YoloDetect::ObjInfos objs;
|
std::vector<ScoredRect> m_objs;
|
// TrackingTrigger *trackingTrigger;
|
std::vector<cv::Mat> triggerMats;
|
::YoloDetect::stringData cocoData;
|
FastFdsWithLock *fdfsClient;
|
|
TriggerElement m_triggerElement;
|
SdkRule m_sdkRule;
|
};
|
|
#endif // YOLORPCELEMENT_H
|
