annotate.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 <cstdint>
#include <filesystem>
#include <optional>
#include <string>
#include <vector>
 
namespace reusex::vision {

struct AnnotationConfig {
  bool use_cuda = false;  
  size_t batch_size = 16; 
  bool shuffle = false;   
  size_t num_workers = 4; 
  size_t prefetch_batches =
      8; 
  bool skip_annotated = false; 
  float confidence = 0.5f;     
  std::vector<std::string>
      prompts; 
  std::optional<uint32_t> seed =
      42; 
  bool video = false; 
};

auto annotate(const std::filesystem::path &dbPath,
              const std::filesystem::path &modelPath,
              const AnnotationConfig &config = AnnotationConfig{}) -> int;

inline bool is_sequence_boundary(std::size_t index, int node_id, int prev_id) {
  return index == 0 || node_id <= prev_id;
}
 
} // namespace reusex::vision