Backend.hpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2026 Povl Filip Sonne-Frederiksen
//
// SPDX-License-Identifier: GPL-3.0-or-later
 
#pragma once
#include "reusex/vision/IMLBackend.hpp"
 
namespace reusex::vision::onnx {

class ONNXBackend : public IMLBackend {
    public:
  ONNXBackend() = default;

  std::unique_ptr<IModel>
  create_model(const Model type,
               const std::filesystem::path &modelPath,
               bool use_cuda = false) override;

  std::unique_ptr<IDataset>
  create_dataset(const std::filesystem::path &datasetPath) override;
};
 
} // namespace reusex::vision::onnx