NeuralNetwork.hpp

Go to the documentation of this file.

#pragma once
 
#include "depthai/openvino/OpenVINO.hpp"
#include "depthai/pipeline/Node.hpp"
 
// standard
#include <fstream>
 
// shared
#include <depthai-shared/properties/NeuralNetworkProperties.hpp>
 
namespace dai {
namespace node {
 
class NeuralNetwork : public NodeCRTP<Node, NeuralNetwork, NeuralNetworkProperties> {
   public:
    constexpr static const char* NAME = "NeuralNetwork";
 
   protected:
    tl::optional<OpenVINO::Version> getRequiredOpenVINOVersion() override;
    tl::optional<OpenVINO::Version> networkOpenvinoVersion;
 
   public:
    NeuralNetwork(const std::shared_ptr<PipelineImpl>& par, int64_t nodeId);
    NeuralNetwork(const std::shared_ptr<PipelineImpl>& par, int64_t nodeId, std::unique_ptr<Properties> props);
 
    Input input{*this, "in", Input::Type::SReceiver, true, 5, true, {{DatatypeEnum::Buffer, true}}};
 
    Output out{*this, "out", Output::Type::MSender, {{DatatypeEnum::NNData, false}}};
 
    Output passthrough{*this, "passthrough", Output::Type::MSender, {{DatatypeEnum::Buffer, true}}};
 
    InputMap inputs;
 
    OutputMap passthroughs;
 
    // Specify local filesystem path to load the blob (which gets loaded at loadAssets)
    void setBlobPath(const dai::Path& path);
 
    void setBlob(OpenVINO::Blob blob);
 
    void setBlob(const dai::Path& path);
 
    void setNumPoolFrames(int numFrames);
 
    void setNumInferenceThreads(int numThreads);
 
    void setNumNCEPerInferenceThread(int numNCEPerThread);
 
    int getNumInferenceThreads();
    // TODO add getters for other API
};
 
}  // namespace node
}  // namespace dai