iVS3D/_neural_net_factory_8h_source.html

#pragma once


#include "NeuralNet.h"

#include "NeuralError.h"


#include <memory>

#include <string>

#include <iostream>

#include <tl/expected.hpp>


namespace NN

{


    class NeuralNetFactory

    {

    public:

        static tl::expected<NeuralNetPtr, NeuralError> create(const std::string& modelPath, bool useGpu = false, int gpuId = 0);

    };


}

NeuralError.h
Defines error handling classes for the neural network module.

NeuralNet.h
Contains the NeuralNet interface for neural network inference.

NN::NeuralNetFactory
Factory class for creating NeuralNet instances. This abstracts from the underlying inference engine (...
Definition NeuralNetFactory.h:26

NN::NeuralNetFactory::create
static tl::expected< NeuralNetPtr, NeuralError > create(const std::string &modelPath, bool useGpu=false, int gpuId=0)
Create a NeuralNet instance from a model file.
Definition NeuralNetFactory.cpp:4

NN
NN Neural Network Library containing Tensor and NeuralNet classes for inference.
Definition NeuralError.h:13