ReUseX  0.0.5
3D Point Cloud Processing for Building Reuse
Loading...
Searching...
No Matches
Dataset.hpp
Go to the documentation of this file.
1// SPDX-FileCopyrightText: 2025 Povl Filip Sonne-Frederiksen
2//
3// SPDX-License-Identifier: GPL-3.0-or-later
4
5#pragma once
6#include "reusex/vision/IData.hpp"
7#include "reusex/vision/IDataset.hpp"
8#include "reusex/vision/libtorch/Data.hpp"
9
10namespace reusex::vision::libtorch {
11
18class LibTorchDataset : public IDataset {
19 public:
20 using IDataset::IDataset;
21
26 IDataset::Pair get(const std::size_t index) const override;
27
32 bool save(const std::span<Pair> &data) override;
33};
34
35} // namespace reusex::vision::libtorch
reusex::vision::IDataset::IDataset
IDataset(std::shared_ptr< ProjectDB > database)
reusex::vision::IDataset::Pair
std::pair< std::unique_ptr< IData >, size_t > Pair
Definition IDataset.hpp:43
reusex::vision::libtorch::LibTorchDataset
LibTorch dataset for RTABMap databases.
Definition Dataset.hpp:18
reusex::vision::libtorch::LibTorchDataset::get
IDataset::Pair get(const std::size_t index) const override
Get a data sample at the given index.
reusex::vision::libtorch::LibTorchDataset::IDataset
IDataset(std::shared_ptr< ProjectDB > database)
reusex::vision::libtorch::LibTorchDataset::save
bool save(const std::span< Pair > &data) override
Save processed results back to the database.