class

GS::MASt3R::PointmapHead

Inherits Reference < Object

Pointmap head: converts features to 3D points with confidence

Constructors

new(in_features : Int32, device : Tensor::Device = Tensor::Device::GPU)
Source

Instance methods

conf_head
Source
forward(features : Autograd::Variable) : Tuple(Autograd::Variable, Autograd::Variable)

Forward pass features: [batch, height, width, in_features] Returns: {xyz: [batch, height, width, 3], conf: [batch, height, width, 1]}

Source
parameters
Source
xyz_head
Source