pub struct MaskDecoder { /* private fields */ }
Implementations§
source§impl MaskDecoder
impl MaskDecoder
pub fn new( transformer_dim: usize, num_multimask_outputs: usize, iou_head_depth: usize, iou_head_hidden_dim: usize, vb: VarBuilder<'_> ) -> Result<Self>
pub fn forward( &self, image_embeddings: &Tensor, image_pe: &Tensor, sparse_prompt_embeddings: &Tensor, dense_prompt_embeddings: &Tensor, multimask_output: bool ) -> Result<(Tensor, Tensor)>
Trait Implementations§
Auto Trait Implementations§
impl Freeze for MaskDecoder
impl !RefUnwindSafe for MaskDecoder
impl Send for MaskDecoder
impl Sync for MaskDecoder
impl Unpin for MaskDecoder
impl !UnwindSafe for MaskDecoder
Blanket Implementations§
source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more