equiadapt
- equiadapt package
- Subpackages
- Module contents
BaseCanonicalization
ContinuousGroupCanonicalization
ContinuousGroupCanonicalization.canonicalization_network
ContinuousGroupCanonicalization.beta
ContinuousGroupCanonicalization.__init__()
ContinuousGroupCanonicalization.canonicalizationnetworkout_to_groupelement()
ContinuousGroupCanonicalization.canonicalize()
ContinuousGroupCanonicalization.invert_canonicalization()
ContinuousGroupCanonicalization.get_prior_regularization_loss()
ContinuousGroupCanonicalization.get_identity_metric()
ContinuousGroupCanonicalization.canonicalizationnetworkout_to_groupelement()
ContinuousGroupCanonicalization.canonicalize()
ContinuousGroupCanonicalization.get_identity_metric()
ContinuousGroupCanonicalization.get_prior_regularization_loss()
ContinuousGroupCanonicalization.invert_canonicalization()
ContinuousGroupImageCanonicalization
ContinuousGroupImageCanonicalization.__init__()
ContinuousGroupImageCanonicalization.get_rotation_matrix_from_vector()
ContinuousGroupImageCanonicalization.get_groupelement()
ContinuousGroupImageCanonicalization.transformations_before_canonicalization_network_forward()
ContinuousGroupImageCanonicalization.get_group_from_out_vectors()
ContinuousGroupImageCanonicalization.canonicalize()
ContinuousGroupImageCanonicalization.invert_canonicalization()
ContinuousGroupImageCanonicalization.canonicalize()
ContinuousGroupImageCanonicalization.get_group_from_out_vectors()
ContinuousGroupImageCanonicalization.get_groupelement()
ContinuousGroupImageCanonicalization.invert_canonicalization()
ContinuousGroupImageCanonicalization.transformations_before_canonicalization_network_forward()
ContinuousGroupPointcloudCanonicalization
ConvNetwork
CustomEquivariantNetwork
DiscreteGroupCanonicalization
DiscreteGroupCanonicalization.canonicalization_network
DiscreteGroupCanonicalization.beta
DiscreteGroupCanonicalization.gradient_trick
DiscreteGroupCanonicalization.__init__()
DiscreteGroupCanonicalization.groupactivations_to_groupelementonehot()
DiscreteGroupCanonicalization.canonicalize()
DiscreteGroupCanonicalization.invert_canonicalization()
DiscreteGroupCanonicalization.get_prior_regularization_loss()
DiscreteGroupCanonicalization.get_identity_metric()
DiscreteGroupCanonicalization.canonicalize()
DiscreteGroupCanonicalization.get_identity_metric()
DiscreteGroupCanonicalization.get_prior_regularization_loss()
DiscreteGroupCanonicalization.groupactivations_to_groupelementonehot()
DiscreteGroupCanonicalization.invert_canonicalization()
DiscreteGroupImageCanonicalization
DiscreteGroupImageCanonicalization.__init__()
DiscreteGroupImageCanonicalization.groupactivations_to_groupelement()
DiscreteGroupImageCanonicalization.get_groupelement()
DiscreteGroupImageCanonicalization.transformations_before_canonicalization_network_forward()
DiscreteGroupImageCanonicalization.canonicalize()
DiscreteGroupImageCanonicalization.invert_canonicalization()
DiscreteGroupImageCanonicalization.canonicalize()
DiscreteGroupImageCanonicalization.get_group_activations()
DiscreteGroupImageCanonicalization.get_groupelement()
DiscreteGroupImageCanonicalization.groupactivations_to_groupelement()
DiscreteGroupImageCanonicalization.invert_canonicalization()
DiscreteGroupImageCanonicalization.transformations_before_canonicalization_network_forward()
ESCNNEquivariantNetwork
ESCNNSteerableNetwork
ESCNNWRNEquivariantNetwork
ESCNNWideBasic
ESCNNWideBottleneck
EquivariantPointcloudCanonicalization
GroupEquivariantImageCanonicalization
IdentityCanonicalization
IdentityCanonicalization.canonicalization_network
IdentityCanonicalization.__init__()
IdentityCanonicalization.canonicalize()
IdentityCanonicalization.canonicalize()
IdentityCanonicalization.get_identity_metric()
IdentityCanonicalization.get_prior_regularization_loss()
IdentityCanonicalization.invert_canonicalization()
LieParameterization
OptimizedGroupEquivariantImageCanonicalization
OptimizedGroupEquivariantImageCanonicalization.__init__()
OptimizedGroupEquivariantImageCanonicalization.rotate_and_maybe_reflect()
OptimizedGroupEquivariantImageCanonicalization.group_augment()
OptimizedGroupEquivariantImageCanonicalization.get_group_activations()
OptimizedGroupEquivariantImageCanonicalization.get_optimization_specific_loss()
OptimizedGroupEquivariantImageCanonicalization.get_group_activations()
OptimizedGroupEquivariantImageCanonicalization.get_optimization_specific_loss()
OptimizedGroupEquivariantImageCanonicalization.group_augment()
OptimizedGroupEquivariantImageCanonicalization.rotate_and_maybe_reflect()
OptimizedSteerableImageCanonicalization
OptimizedSteerableImageCanonicalization.__init__()
OptimizedSteerableImageCanonicalization.get_rotation_matrix_from_vector()
OptimizedSteerableImageCanonicalization.group_augment()
OptimizedSteerableImageCanonicalization.get_groupelement()
OptimizedSteerableImageCanonicalization.get_optimization_specific_loss()
OptimizedSteerableImageCanonicalization.get_groupelement()
OptimizedSteerableImageCanonicalization.get_optimization_specific_loss()
OptimizedSteerableImageCanonicalization.get_rotation_matrix_from_vector()
OptimizedSteerableImageCanonicalization.group_augment()
ResNet18Network
RotationEquivariantConv
RotationEquivariantConvLift
RotoReflectionEquivariantConv
RotoReflectionEquivariantConvLift
SteerableImageCanonicalization
VNBatchNorm
VNBilinear
VNLeakyReLU
VNLinear
VNLinearLeakyReLU
VNMaxPool
VNSmall
VNSoftplus
VNStdFeature
get_action_on_image_features()
get_graph_feature_cross()
gram_schmidt()