#ifndef PAYOLORPCELEMENT_H
|
#define PAYOLORPCELEMENT_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 "../StructureApp/TrackingTrigger.h"
|
#include <QtCore/QVector>
|
#include <QtCore/QJsonArray>
|
#include <QtGui/qpolygon.h>
|
class QSharedMemory;
|
class ControllerConfig;
|
class PaYoloRpcElement : public basic::PipeElement {
|
public:
|
PaYoloRpcElement(string);
|
|
~PaYoloRpcElement();
|
|
void setImage(const cv::Mat &value);
|
|
int getRealNum() const;
|
void SetRuleCfg(const ControllerConfig* cfg);
|
|
::YoloDetect::ObjInfos getObjects() const;
|
|
std::vector<cv::Mat> getTriggerMats();
|
|
std::vector<ScoredRect> getLastScoreRects() const;
|
|
|
|
void setFdfs(FastFdsWithLock *p_fdfsClient) {
|
fdfsClient = p_fdfsClient;
|
}
|
|
private:
|
virtual void threadFunc() override;
|
bool isInWeekRuleTime();
|
QJsonArray getJsonArrayFromQString(const QString strJson);
|
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;
|
cv::Mat mask,numMask;
|
QString mStrArea;
|
QPolygon mPolygon;
|
int mRealNum;
|
ControllerConfig* mCfg;
|
|
};
|
|
#endif // YOLORPCELEMENT_H
|
