segment_frame.hpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2026 Povl Filip Sonne-Frederiksen
//
// SPDX-License-Identifier: GPL-3.0-or-later
 
#pragma once
 
#include "../global-params.hpp"
 
#include <CLI/CLI.hpp>
 
#include <filesystem>
#include <string>
#include <vector>
 
struct SubcommandSegmentFrameOptions {
  std::filesystem::path net_path;
  int frame_id = -1;
  std::vector<std::string> texts;
  std::vector<std::string> boxes;
  std::vector<std::string> points;
  float confidence = 0.5f;
  bool use_cuda = false;
  bool save = true;
};
 
void setup_subcommand_create_segment_frame(CLI::App &app,
                                           std::shared_ptr<RuxOptions> opt);
 
int run_subcommand_create_segment_frame(
    const SubcommandSegmentFrameOptions &opt, const RuxOptions &global_opt);