ReUseX  0.0.5
3D Point Cloud Processing for Building Reuse
Loading...
Searching...
No Matches
Sam3Dataset.hpp
Go to the documentation of this file.
1// SPDX-FileCopyrightText: 2026 Povl Filip Sonne-Frederiksen
2//
3// SPDX-License-Identifier: GPL-3.0-or-later
4
5#pragma once
6#include "reusex/vision/IDataset.hpp"
7#include "reusex/vision/onnx/Sam3Data.hpp"
8
9namespace reusex::vision::onnx {
10
15class ONNXSam3Dataset : public IDataset {
16 public:
17 using IDataset::IDataset;
18
22 IDataset::Pair get(const std::size_t index) const override;
23
27 bool save(const std::span<Pair> &data) override;
28
29 private:
30 bool class_map_saved_ = false;
31};
32
33} // namespace reusex::vision::onnx
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::onnx::ONNXSam3Dataset
Dataset for ONNX Runtime SAM3 model inference.
Definition Sam3Dataset.hpp:15
reusex::vision::onnx::ONNXSam3Dataset::IDataset
IDataset(std::shared_ptr< ProjectDB > database)
reusex::vision::onnx::ONNXSam3Dataset::save
bool save(const std::span< Pair > &data) override
Save label images from completed inference back to the database.
reusex::vision::onnx::ONNXSam3Dataset::get
IDataset::Pair get(const std::size_t index) const override
Load an image at the given index into an ONNXSam3Data.