segment_image.hpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2026 Povl Filip Sonne-Frederiksen
//
// SPDX-License-Identifier: GPL-3.0-or-later
//
// Single-frame SAM3 segmentation — the per-image building block factored out
// of segment_panorama so the GUI endpoint and CLI can call it directly (#409).
//
// Backend dispatch: tries TensorRT first (most common), then ONNX Runtime.
// Detection is runtime, via the IData subtype returned by forward(). At least
// one of REUSEX_USE_TENSORRT / REUSEX_USE_ONNX_RUNTIME must be compiled into
// the vision library for any result to be produced.
 
#pragma once
 
#include "reusex/vision/IModel.hpp"
#include "reusex/vision/sam3_prompt.hpp"
 
#include <opencv2/core.hpp>
 
#include <cstddef>
#include <vector>
 
namespace reusex::vision {

std::size_t clip_labels_to_prompt_boxes(cv::Mat &labels,
                                        const std::vector<Sam3Prompt> &prompts);

struct SegmentImageInfo {
  bool geometry_prompts_used = false;
};

cv::Mat segment_image(IModel &model, const cv::Mat &image_bgr,
                      const std::vector<Sam3Prompt> &prompts = {},
                      float confidence = 0.5f,
                      SegmentImageInfo *info = nullptr);
 
} // namespace reusex::vision