Backend.hpp

Go to the documentation of this file.

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

class LibTorchBackend : public IMLBackend {
    public:
  LibTorchBackend() = 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::libtorch