sam_seg_module.cpp File Reference

#include "ed_sensor_integration/kinect/segmodules/sam_seg_module.h"
#include <filesystem>
#include <utility>
#include <sam_onnx_ros/segmentation.hpp>
#include <sensor_msgs/Image.h>
#include <sensor_msgs/PointCloud2.h>
#include <yolo_onnx_ros/detection.hpp>

Include dependency graph for sam_seg_module.cpp:

Go to the source code of this file.

Functions
void	overlayMasksOnImage_ (cv::Mat &rgb, const std::vector< cv::Mat > &masks)
	Overlay segmentation masks on the RGB image for visualization purposes. More...
SegmentationResult	SegmentationPipeline (const cv::Mat &img, tue::Configuration &config)
	Segmentation pipeline that processes the input image and generates segmentation masks. More...

Function Documentation

overlayMasksOnImage_()

void overlayMasksOnImage_	(	cv::Mat &	rgb,
		const std::vector< cv::Mat > &	masks
	)

Overlay segmentation masks on the RGB image for visualization purposes.

Parameters

rgb	The RGB image to overlay masks on.
masks	The segmentation masks to overlay.

Definition at line 69 of file sam_seg_module.cpp.

SegmentationPipeline()

SegmentationResult SegmentationPipeline	(	const cv::Mat &	img,
		tue::Configuration &	config
	)

Segmentation pipeline that processes the input image and generates segmentation masks.

Parameters

img	The input RGB image to segment.
config	Configuration containing model paths.

Returns

SegmentationResult The generated segmentation masks, bounding boxes, labels, and confidences.

Definition at line 11 of file sam_seg_module.cpp.