Struct MaskDecoder 

pub struct MaskDecoder { /* private fields */ }

Decodes image features + prompt embeddings into segmentation masks.

Implementations

impl MaskDecoder

pub fn new(config: &SAMConfig) -> Self

Create a new mask decoder.

pub fn decode( &self, image_features: &Array2<f64>, prompt_embedding: &PromptEmbedding, image_size: (usize, usize), ) -> Result<SegmentationResult>

Decode masks from image features and prompt embeddings.

Arguments

• image_features - Encoder output [spatial_tokens, channels].
• prompt_embedding - Output of the prompt encoder.
• image_size - Original image (height, width).

Returns

A SegmentationResult with num_mask_outputs candidate masks at the original image resolution, together with IoU predictions and stability scores.

Trait Implementations

impl Clone for MaskDecoder

fn clone(&self) -> MaskDecoder

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for MaskDecoder

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.