CscDetectorEngineHandPose.h

Go to the documentation of this file.

//
//  CscDetectorEngineMire.hpp
//  ConscienceRobotManager
//
//  Created by Victor Coru on 26/05/2020.
//  Copyright © 2020 Conscience Robotics. All rights reserved.
//
 
#ifndef CscDetectorEngineHandPose_h
#define CscDetectorEngineHandPose_h
 
#include "CoreObjects/CscWorldObject.h"
#include "DetectorEngine/CscDetectorEngine.h"
#include "DetectorEngine/CscDNNPool.h"
#include <ctime>
#include <opencv2/opencv.hpp>
#include <vector>
 
using std::string,std::vector;
using cv::dnn::Net;
 
 
namespace conscience_core::detector_engine {
 
    using SafeNeuralNetwork = CscDNNPool::SafeNeuralNetwork;
 
    class CscDetectorEngineHandPose : public CscDetectorEngine {
    public:
        CscDetectorEngineHandPose(ptr<CscDNNPool> dnnPool,
                                  const string & protoFile,
                                  const string & weightsFile,
                                  bool drawPredictionOnImage);
 
        ptr<DetectionResult> detectOnImage(
            const DetectorSourceImage &image, ptr<DetectionParameters> parameters) override;
 
        virtual bool equals(CscDetectorEngine *) const override;
 
    private:
        const string protoFile;
        const string weightsFile;
        bool drawPredictionOnImage;
        cv::Scalar detectedRectanglesColor = cv::Scalar(100, 200, 10);
        
        ptr<SafeNeuralNetwork> neuralNetwork;
        
        unique_ptr<CscLogger> logger;
    };
 
}
 
#endif