Struct candle_transformers::models::segment_anything::prompt_encoder::PromptEncoder

pub struct PromptEncoder { /* private fields */ }

Implementations

impl PromptEncoder

pub fn new( embed_dim: usize, image_embedding_size: (usize, usize), input_image_size: (usize, usize), mask_in_chans: usize, vb: VarBuilder<'_> ) -> Result<Self>

pub fn get_dense_pe(&self) -> Result<Tensor>

pub fn forward( &self, points: Option<(&Tensor, &Tensor)>, boxes: Option<&Tensor>, masks: Option<&Tensor> ) -> Result<(Tensor, Tensor)>

Trait Implementations

impl Debug for PromptEncoder

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

Formats the value using the given formatter.