#ifndef IMAGEDRAWELEMENT_H
|
#define IMAGEDRAWELEMENT_H
|
|
#include <YoloServer.h>
|
//#include <FaceServer.h>
|
#include <basic/pipe_element/ImageFactoryElement.h>
|
#include <basic/core/DoubleBufferedData.h>
|
#include "../StructureApp/TrackingTrigger.h"
|
|
class ImageDrawElement : public ImageFactoryElement {
|
public:
|
ImageDrawElement();
|
|
// ImageFactoryElement interface
|
// void setFaces(std::vector<ScoredRect> value);
|
|
void setYoloObjects(std::vector<ScoredRect> value);
|
|
private:
|
virtual void processImage(cv::Mat &) override;
|
|
void darwProperty(cv::Mat &image, string key, string value, int x, int y);
|
|
private:
|
// DoubleBufferedData<std::vector<ScoredRect>> faces;
|
DoubleBufferedData<std::vector<ScoredRect>> yoloObjects;
|
};
|
|
|
#endif // IMAGEDRAWELEMENT_H
|
