#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 <RapidStructureApp/TriggerElement.h>
|
#include "TrackingTrigger.h"
|
#include "DBStruct.h"
|
|
class QSharedMemory;
|
|
class YoloRpcElement : public basic::PipeElement {
|
public:
|
YoloRpcElement(std::string shareMemoryName);
|
|
~YoloRpcElement();
|
|
void setImage(const cv::Mat &value);
|
|
cv::Mat getImage() const;
|
|
::YoloDetect::ObjInfos getObjects() const;
|
|
std::vector<ScoredRect> getLastScoreRects() const;
|
|
private:
|
virtual void threadFunc() override;
|
|
private:
|
IceRpcClient<YoloDetect::YoloDetectServerPrx> rpcClient;
|
cv::Mat image;
|
QSharedMemory *sharedMemory;
|
::YoloDetect::ObjInfos objs;
|
::YoloDetect::stringData cocoData;
|
std::vector<ScoredRect> triggerScoredRects;
|
TrackingTrigger *trackingTrigger;
|
|
};
|
|
#endif // YOLORPCELEMENT_H
|
