ReUseX  0.0.5
3D Point Cloud Processing for Building Reuse
Loading...
Searching...
No Matches
annotate.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 <filesystem>
7
8namespace reusex::vision {
9
11struct AnnotationConfig {
12 bool use_cuda = false;
13 size_t batch_size = 16;
14 bool shuffle = false;
15 size_t num_workers = 4;
16 size_t prefetch_batches = 8;
17};
18
27auto annotate(const std::filesystem::path &dbPath,
28 const std::filesystem::path &modelPath,
29 const AnnotationConfig &config = AnnotationConfig{}) -> int;
30
31} // namespace reusex::vision
reusex::vision::annotate
auto annotate(const std::filesystem::path &dbPath, const std::filesystem::path &modelPath, const AnnotationConfig &config=AnnotationConfig{}) -> int
Run semantic annotation on sensor frames using ML models.
reusex::vision::AnnotationConfig
Configuration for ML annotation inference and data loading.
Definition annotate.hpp:11
reusex::vision::AnnotationConfig::use_cuda
bool use_cuda
Force CUDA acceleration.
Definition annotate.hpp:12
reusex::vision::AnnotationConfig::num_workers
size_t num_workers
Number of worker threads (recommended: 2-4).
Definition annotate.hpp:15
reusex::vision::AnnotationConfig::shuffle
bool shuffle
Shuffle dataset before processing.
Definition annotate.hpp:14
reusex::vision::AnnotationConfig::batch_size
size_t batch_size
Batch size for inference (recommended: 8-64).
Definition annotate.hpp:13
reusex::vision::AnnotationConfig::prefetch_batches
size_t prefetch_batches
Batches to prefetch (recommended: 2-3x workers).
Definition annotate.hpp:16