Struct MaskHead 

pub struct MaskHead { /* private fields */ }

Mask prediction head for instance segmentation (Mask R-CNN style).

Takes RoI-pooled features [N, H, W, C] and produces binary masks [N, mask_h, mask_w, num_classes] via a series of conv layers + upsample.

Implementations

impl MaskHead

pub fn new( in_channels: usize, channels: usize, num_classes: usize, mask_size: usize, num_conv: usize, ) -> Result<Self, ModelError>

Create a mask head with num_conv intermediate conv layers.

pub fn num_classes(&self) -> usize

pub fn mask_size(&self) -> usize

pub fn channels(&self) -> usize

pub fn forward_inference( &self, roi_features: &Tensor, ) -> Result<Tensor, ModelError>

Forward pass: conv layers → ReLU → upsample → class prediction.

Trait Implementations

impl Clone for MaskHead

fn clone(&self) -> MaskHead

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MaskHead

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

Formats the value using the given formatter.

impl PartialEq for MaskHead

fn eq(&self, other: &MaskHead) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for MaskHead