#include <opencv2/core.hpp>#include <geolib/datatypes.h>#include <ros/publisher.h>#include <tue/config/configuration.h>#include "ed/kinect/entity_update.h"#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 | |
| 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... | |
| 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.
