Struct DeformableConv2dLayer 

pub struct DeformableConv2dLayer {
    pub weight: Tensor,
    pub offset_weight: Tensor,
    pub bias: Option<Tensor>,
    pub stride: usize,
    pub padding: usize,
    pub kernel_h: usize,
    pub kernel_w: usize,
    /* private fields */
}

Deformable 2D convolution layer (NHWC layout).

Like standard conv2d but sampling positions are offset by learned offsets. An internal offset convolution produces offsets from the input, then those offsets are used to sample input with bilinear interpolation.

Weight shape: [kH, kW, C_in, C_out]. Offset weight shape: [kH, kW, C_in, kH*kW*2] – conv producing offsets.

Fields

weight: Tensor

offset_weight: Tensor

bias: Option<Tensor>

stride: usize

padding: usize

kernel_h: usize

kernel_w: usize

Implementations

impl DeformableConv2dLayer

pub fn new( in_channels: usize, out_channels: usize, kernel_h: usize, kernel_w: usize, stride: usize, padding: usize, weight: Tensor, offset_weight: Tensor, bias: Option<Tensor>, ) -> Result<Self, ModelError>

pub fn zero_init( in_channels: usize, out_channels: usize, kernel_h: usize, kernel_w: usize, stride: usize, padding: usize, use_bias: bool, ) -> Result<Self, ModelError>

pub fn in_channels(&self) -> usize

pub fn out_channels(&self) -> usize

pub fn kernel_h(&self) -> usize

pub fn kernel_w(&self) -> usize

pub fn stride(&self) -> usize

pub fn padding(&self) -> usize

pub fn weight(&self) -> &Tensor

pub fn offset_weight(&self) -> &Tensor

pub fn bias(&self) -> Option<&Tensor>

pub fn weight_node(&self) -> Option<NodeId>

pub fn register_params(&mut self, graph: &mut Graph)

pub fn forward( &self, graph: &mut Graph, input: NodeId, ) -> Result<NodeId, ModelError>

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

Trait Implementations

impl Clone for DeformableConv2dLayer

fn clone(&self) -> DeformableConv2dLayer

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DeformableConv2dLayer

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

Formats the value using the given formatter.

impl PartialEq for DeformableConv2dLayer

fn eq(&self, other: &DeformableConv2dLayer) -> 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 DeformableConv2dLayer