NeuralBase.h

Go to the documentation of this file.

#pragma once
 
#include "Detection.h"
#include <opencv2/dnn.hpp>
 
namespace core {
 
    class NeuralBase {
    public:
        NeuralBase() = default;
 
        std::string getName();
 
        double getInferenceTime();
 
        void loadNetwork(
            std::string framework,
            std::string modelWeightsFile,
            std::string classesFile = "",
            std::string modelConfigurationFile = "",
            std::string inputLayerName = "",
            std::string outputLayerName = ""
        );
 
        void configureNetwork(std::string name,
            //Preprocessor params
            int inputWidth,
            int inputHeight,
            double scale,
            double mean0,
            double mean1,
            double mean2,
            //Processing params
            std::string configBackend,
            std::string configTarget,
            //Postprocessing params
            double confidenceThreshold);
 
        std::vector<std::string> getOutputsNames(const cv::dnn::Net& net);
 
    protected:
 
        void readClassNames(const std::string& filePath);
 
        std::string mFramework;
        std::string mInputLayerName;
        std::string mOutputLayerName;
        std::string mName;
 
        // Loaded model
        cv::dnn::Net mNet;
 
        // Object class names
        std::vector<std::string> classes;
 
        // Initialize the parameters
 
        // Preprocessor params:
        double scaleFactor = 1.0;
        cv::Scalar mean{ 0.0, 0.0, 0.0 }; // Mean of network's input image
        cv::Size size{ 128, 128 }; // Width and Height of network's input image
 
        // Processor params:
        cv::dnn::Backend selectedBackend =
            cv::dnn::DNN_BACKEND_OPENCV; // Select preferable backend
 
        cv::dnn::Target selectedTarget =
            cv::dnn::DNN_TARGET_CPU; // Select preferable target
 
        // Postprocessor params:
        float confThreshold = 0.5F; // Confidence threshold
 
        // Choosable processor params
        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} 
        };
 
        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 core