FaceDetectorYunet.h

#pragma once
 
#include "Detection.h"
#include <opencv2/dnn.hpp>
#include <opencv2/objdetect/face.hpp>
 
namespace core {
    namespace face {
 
        class FaceDetectorYunet {
        public:
            FaceDetectorYunet() = default;
 
            ~FaceDetectorYunet() = 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 = 0.9,
                double nmsThreshold = 0.3,
                double maxWH = 312.0,
                int topK = 5000);
 
            std::vector<core::Detection> process(const cv::Mat& frame);
 
        private:
            cv::Ptr<cv::FaceDetectorYN> mDetector; 
            double maxWidthOrHeight; 
            // Available backend configurations for DNN processing.
            std::map<std::string, cv::dnn::Backend> backends = {
                {"DNN_BACKEND_DEFAULT", cv::dnn::DNN_BACKEND_DEFAULT},
                {"DNN_BACKEND_HALIDE", cv::dnn::DNN_BACKEND_HALIDE},
                {"DNN_BACKEND_INFERENCE_ENGINE", cv::dnn::DNN_BACKEND_INFERENCE_ENGINE},
                {"DNN_BACKEND_OPENCV", cv::dnn::DNN_BACKEND_OPENCV},
                {"DNN_BACKEND_CUDA", cv::dnn::DNN_BACKEND_CUDA}
            };
 
            // Available target configurations for DNN processing.
            std::map<std::string, cv::dnn::Target> targets = {
                {"DNN_TARGET_CPU", cv::dnn::DNN_TARGET_CPU},
                {"DNN_TARGET_CUDA", cv::dnn::DNN_TARGET_CUDA},
                {"DNN_TARGET_OPENCL", cv::dnn::DNN_TARGET_OPENCL},
                {"DNN_TARGET_OPENCL_FP16", cv::dnn::DNN_TARGET_OPENCL_FP16},
                {"DNN_TARGET_MYRIAD", cv::dnn::DNN_TARGET_MYRIAD}
            };
        };
 
    } // namespace face
} // namespace core