| | |
| | | #ifndef IMAGEDRAWELEMENT_H |
| | | #define IMAGEDRAWELEMENT_H |
| | | #include <basic/pipe_element/ImageFactoryElement.h> |
| | | #include <mutex> |
| | | |
| | | class ImageDrawElement : public ImageFactoryElement |
| | | { |
| | | #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(); |
| | | void setRects(const std::vector<cv::Rect2f> &value); |
| | | |
| | | // ImageFactoryElement interface |
| | | // void setFaces(std::vector<ScoredRect> value); |
| | | |
| | | void setYoloObjects(std::vector<ScoredRect> value); |
| | | |
| | | private: |
| | | std::mutex rectsMtx; |
| | | std::vector<cv::Rect2f> rects; |
| | | std::vector<cv::Rect2f> rectsBuffer; |
| | | bool rectsUpdated; |
| | | virtual void processImage(cv::Mat& image); |
| | | 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 |