class

GS::MASt3R::DPTHead

Inherits Reference < Object

DPT Head for MASt3R Outputs 3D points + confidence (4 channels)

Constructors

new(encoder_dim : Int32 = 1024, decoder_dim : Int32 = 768, device : Tensor::Device = Tensor::Device::GPU)
Source

Instance methods

act_postprocess
Source
forward(decoder_out : Autograd::Variable, patch_h : Int32, patch_w : Int32) : Tuple(Autograd::Variable, Autograd::Variable)

Forward pass - simplified version using final decoder output decoder_out: [batch, seq_len, embed_dim] Returns: {points: [batch, H, W, 3], conf: [batch, H, W, 1]} Note: H, W are at patch resolution (e.g., 32x32 for 512px/16patch)

Source
head
Source
local_features_fc1
Source
local_features_fc2
Source
parameters
Source
refinenets
Source
scratch_layers
Source