Dataset.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/IData.hpp"
#include "reusex/vision/IDataset.hpp"
#include "reusex/vision/libtorch/Data.hpp"
 
namespace reusex::vision::libtorch {

class LibTorchDataset : public IDataset {
    public:
  using IDataset::IDataset;

  IDataset::Pair get(const std::size_t index) const override;

  bool save(const std::span<Pair> &data) override;
};
 
} // namespace reusex::vision::libtorch