ReUseX  0.0.5
3D Point Cloud Processing for Building Reuse
Loading...
Searching...
No Matches
reusex::vision::onnx::ONNXBackend Class Reference

ONNX Runtime backend implementation of the IMLBackend interface. More...

#include <Backend.hpp>

Inheritance diagram for reusex::vision::onnx::ONNXBackend:
Collaboration diagram for reusex::vision::onnx::ONNXBackend:

Public Member Functions

 ONNXBackend ()=default
std::unique_ptr< IModelcreate_model (const Model type, const std::filesystem::path &modelPath, bool use_cuda=false) override
 Create an ONNX Runtime model.
std::unique_ptr< IDatasetcreate_dataset (const std::filesystem::path &datasetPath) override
 Create an ONNX dataset from a database path.
Public Member Functions inherited from reusex::vision::IMLBackend
virtual ~IMLBackend ()=default

Detailed Description

ONNX Runtime backend implementation of the IMLBackend interface.

Provides factory methods for creating ONNX Runtime-based models and datasets for use with the backend-agnostic annotation pipeline.

Definition at line 14 of file Backend.hpp.

Constructor & Destructor Documentation

◆ ONNXBackend()

reusex::vision::onnx::ONNXBackend::ONNXBackend ( )
default

Member Function Documentation

◆ create_dataset()

std::unique_ptr< IDataset > reusex::vision::onnx::ONNXBackend::create_dataset ( const std::filesystem::path & datasetPath)
overridevirtual

Create an ONNX dataset from a database path.

Parameters
datasetPathPath to the ReUseX project database file.
Returns
Unique pointer to the created dataset.

Implements reusex::vision::IMLBackend.

◆ create_model()

std::unique_ptr< IModel > reusex::vision::onnx::ONNXBackend::create_model ( const Model type,
const std::filesystem::path & modelPath,
bool use_cuda = false )
overridevirtual

Create an ONNX Runtime model.

Parameters
typeModel type (currently Sam3).
modelPathPath to directory containing ONNX model files.
use_cudaWhether to use CUDA execution provider (defaults to false).
Returns
Unique pointer to the created model.
Exceptions
std::runtime_errorif model type is not supported.

Implements reusex::vision::IMLBackend.


The documentation for this class was generated from the following file: