DeepSDFStruct.deep_sdf.models#

Classes

DeepSDFModel(decoder, ...)

class DeepSDFStruct.deep_sdf.models.DeepSDFModel(decoder, trained_latent_vectors, device)#

Bases: object

Parameters:

  • decoder (torch.nn.modules.module.Module)

  • trained_latent_vectors (torch.Tensor)

export_libtorch_executable(filename)#

Export the trained decoder model to a TorchScript file for use with LibTorch (C++).

Parameters:

filename (str) – Path where the TorchScript model will be saved (e.g. “decoder.pt”).

Example

>>> model.export_libtorch_executable("decoder.pt")
Example input:  tensor([[...]])
Example Output: tensor([[...]])
# The file "decoder.pt" is now ready for loading in LibTorch.