DeepSDFStruct.pretrained_models#

Functions

get_model(model[, checkpoint, device])

Load a pretrained model by name or enum.

list_available_models()

Classes

PretrainedModels(*values)

class DeepSDFStruct.pretrained_models.PretrainedModels(*values)#

Bases: enum.Enum

AnalyticRoundCross = 'analytic_round_cross'#
ChiAndCross = 'chi_and_cross'#
RoundCross = 'round_cross'#
DeepSDFStruct.pretrained_models.get_model(model, checkpoint='latest', device=None)#

Load a pretrained model by name or enum.

Parameters:

  • model (str | PretrainedModels) – model identifier

  • checkpoint (str) – checkpoint file name (default: ‘latest’)

Return type:

DeepSDFModel

Returns:

Trained PyTorch model

DeepSDFStruct.pretrained_models.list_available_models()#