FaceMesh.h

#pragma once
 
#include "Detection.h"
#include "NeuralBase.h"
#include <opencv2/dnn.hpp>
 
namespace core {
    namespace face {
        class FaceMesh : public NeuralBase {
        public:
            FaceMesh() = default;
 
            ~FaceMesh() = default;
 
            void init(std::string framework,
                std::string modelWeightsFile,
                std::string classesFile,
                std::string modelConfigurationFile,
                std::string inputLayerName,
                std::string outputLayerName,
                std::string networkName,
                int inputWidth,
                int inputHeight,
                double scale,
                double mean0,
                double mean1,
                double mean2,
                std::string configBackend,
                std::string configTarget,
                double confidenceThreshold);
 
            void process(const cv::Mat& frame, std::vector<core::Detection>& detections);
 
        private:
            float processDetection(const cv::Mat& frame, core::Detection& detection);
 
            std::vector<core::Detection> mPerviousDetections; 
            long long mTrackingIdCounter = 0; 
        };
    } // namespace face
} // namespace core