sam_seg_module.h File Reference

#include <opencv2/core.hpp>
#include <geolib/datatypes.h>
#include <ros/publisher.h>
#include <tue/config/configuration.h>
#include "ed/kinect/entity_update.h"
#include <atomic>
#include <memory>
#include <string>
#include <utility>
#include <vector>

Include dependency graph for sam_seg_module.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	SamSegPipeline
	Segmentation pipeline class that maintains YOLO and SAM models state. This prevents initializing heavy models iteratively on every frame. More...
struct	SegmentationLatency
	Latency measurements for the segmentation pipeline subsystems. All times are in milliseconds. More...
struct	SegmentationResult
	Result of the segmentation pipeline containing masks, bounding boxes, and YOLO classification info for each detected object. More...

Functions
void	overlayMasksOnImage_ (cv::Mat &rgb, const std::vector< cv::Mat > &masks)
	Overlay segmentation masks on the RGB image for visualization purposes. 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 191 of file sam_seg_module.cpp.