MoveIt Pro API
Core Behaviors for MoveIt Pro
Loading...
Searching...
No Matches
sam2_automasking.hpp File Reference
#include <filesystem>
#include <future>
#include <memory>
#include <string>
#include <vector>
#include <moveit_pro_ml/onnx_sam2_automasking.hpp>
#include <moveit_studio_behavior_interface/async_behavior_base.hpp>
#include <moveit_studio_vision_msgs/msg/mask2_d.hpp>
#include <sensor_msgs/msg/image.hpp>
#include <tl_expected/expected.hpp>
Include dependency graph for sam2_automasking.hpp:
This graph shows which files directly or indirectly include this file:

Classes

struct  moveit_studio::behaviors::GetMasks2DAutomask
 Segment an image using SAM2 automasking with grid-based point generation. More...
 
struct  moveit_studio::behaviors::GetMasks2DAutomask::InferenceHandle
 Interface for SAM2 automasking inference. More...
 
struct  moveit_studio::behaviors::GetMasks2DAutomask::SAM2AutomaskHandle
 Default implementation using SAM2Automasking. More...
 

Namespaces

namespace  cv
 
namespace  moveit_studio
 
namespace  moveit_studio::behaviors
 

Functions

tl::expected< std::vector< cv::Mat >, std::string > moveit_studio::behaviors::refineAndSplit (const moveit_pro_ml::SAM2Automasking::Result &logits, int original_width, int original_height, float nms_threshold, int min_component_area)
 Refine and split SAM2 masks with NMS, component splitting, and upscaling.