CscDNNPool.h

Go to the documentation of this file.

#ifndef CscDNNPool_H
#define CscDNNPool_H
 
#include <CscCommon.h>
#include <Util/os.h>
 
namespace cv {
class Mat;
}
#if CSC_OS_WINDOWS == 1
namespace cv::dnn::dnn4_v20240521 {
class Net;
}
using cv::dnn::dnn4_v20240521::Net;
#else
namespace cv::dnn::dnn4_v20221220 {
    class Net;
}
using cv::dnn::dnn4_v20221220::Net;
#endif
 
using cv::Mat;
 
namespace conscience_core::detector_engine {
    class CscDNNPool {
    public:
        struct SafeNeuralNetwork {
        public:
            void doWithNet(function<void(Net &net)> action);
 
            Mat *process(Mat &inputBlob);
 
            vector<Mat *> processAllLayers(Mat &inputBlob);
            
            SafeNeuralNetwork(const string &modelPath, optional<string> configPath);
            
            void release();
 
        private:
            const string modelPath;
            Net *net = nullptr;
            mutex dnnProcessMutex;
            unsigned long long lastProcessTimeMillis = 0;
 
            std::unique_ptr<CscLogger> logger;
            vector<string> getOutputsNames();
        };
        CscDNNPool();
        ptr<SafeNeuralNetwork> get(const string &modelPath, optional<string> configPath = {});
        void release(const string &modelPath, optional<string> configPath = {});
        
        // TODO : how to release Net
    private:
        map<string, ptr<SafeNeuralNetwork>> networks;
        mutex getMutex;
        
        std::unique_ptr<CscLogger> logger;
 
        string getNetCacheId(const string &modelPath, optional<string> configPath) const;
    };
}
 
#endif