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 "../global-params.hpp"
 
#include <reusex/vision/annotate.hpp>
 
#include <CLI/CLI.hpp>
#include <cstdint>
#include <filesystem>
#include <memory>
#include <string>
 
namespace fs = std::filesystem;
 
struct SubcommandAnnotateOptions {
 
  fs::path net_path = fs::current_path() / "yolov8x-seg.torchscript";
 
  bool isCuda{false};
 
  // Dataloader configuration
  size_t batch_size = 16; // Batch size for inference (recommended: 8-64)
  bool shuffle = false;   // Shuffle dataset (rarely needed for inference)
  size_t num_workers = 4; // Number of worker threads (recommended: 2-4)
  size_t prefetch_batches =
      8; // Batches to prefetch (recommended: 2-3x workers)
  bool skip_annotated = false; // Skip already-annotated frames (resume mode)
  float confidence = 0.5f;     // Detection confidence threshold [0,1]
  // Open-vocabulary concept prompts = the classes to detect. --prompts is a
  // comma-separated list; --prompts-file is one concept per line (# comments
  // allowed). Empty leaves the model's built-in default class list.
  std::string prompts;      // comma-separated concept list
  std::string prompts_file; // path to a file with one concept per line
  bool video = false;       // Use stateful video-tracker path (SAM 3.1). Forces
                            // ordered single-threaded processing.
 
  // Shuffle RNG seed. Fixed by default so shuffled runs are reproducible
  // (docs/STANDARDS.md §6). --random-seed opts into entropy instead.
  uint32_t seed = 42;
  bool random_seed = false; // Seed from std::random_device (non-deterministic)
 
  // Glass depth filter (batch path only; ignored in --video / SAM3.1 mode).
  bool glass_filter = false; // Detect glass/mirror pixels and emit per-frame
                             // confidence maps consumed by 'rux create clouds'.
  float glass_threshold = reusex::vision::AnnotationConfig{}.glass_threshold;
};
 
// Function declarations.
void setup_subcommand_create_annotate(CLI::App &app,
                                      std::shared_ptr<RuxOptions> global_opt);
int run_subcommand_annotate(SubcommandAnnotateOptions const &opt,
                            const RuxOptions &global_opt);