module

GS::MASt3R

Class methods

create_model(variant : Symbol = :vit_large, device : Tensor::Device = Tensor::Device::GPU) : Model

Convenience function to create model with default config

infer_pointcloud(frames : Array(Video::Frame), weights_path : String = "models/mastr/model.safetensors", device : Tensor::Device = Tensor::Device::CPU, conf_threshold : Float32 = 0.3_f32) : Tuple(Tensor, Tensor)

Convenience function to run inference

Source

load_model(weights_path : String, variant : Symbol = :vit_large, device : Tensor::Device = Tensor::Device::GPU) : Model

Load model from checkpoint

Source

load_safetensors(path : String) : SafetensorsLoader

Convenience function

Source

Class methods

Nested types