base-detection.h

Go to the documentation of this file.

// License: Apache 2.0. See LICENSE file in root directory.
// Copyright(c) 2019 Intel Corporation. All Rights Reserved.

#pragma once

#include <librealsense2/rs.hpp>
#include <rs-vino/openvino-helpers.h>  // include all the InferenceEngine headers
#include <string>


namespace openvino_helpers
{
    /*
        Base class for any type of OpenVINO detection implementation, e.g. object_detection, age_gender_detection, etc.

        Encapsulates an ExecutableNetwork via operator->().
    */
    struct base_detection
    {
        InferenceEngine::ExecutableNetwork net;
        InferenceEngine::InferRequest::Ptr _request;
        std::string topoName;
        std::string pathToModel;
        const size_t maxBatch;
        bool isBatchDynamic;
        const bool isAsync;
        mutable bool enablingChecked;
        mutable bool _enabled;
        const bool doRawOutputMessages;

        base_detection( std::string topoName,
            const std::string &pathToModel,
            int maxBatch, bool isBatchDynamic, bool isAsync,
            bool doRawOutputMessages );

        virtual ~base_detection() = default;

        // Loads the network into the Inference Engine device
        void load_into( InferenceEngine::Core & ie, const std::string & deviceName );

        // Encapsulate the contained ExecutableNetwork
        InferenceEngine::ExecutableNetwork* operator->() { return &net; }

        // Each detector will implement its own reading of the network
        virtual InferenceEngine::CNNNetwork read_network() = 0;
            
        virtual void submit_request();
        virtual void wait();
            
        bool enabled() const;
    };
}