equiadapt

• equiadapt package
  • Subpackages
    • equiadapt.common package
      • Submodules
      • equiadapt.common.basecanonicalization module
      • equiadapt.common.utils module
      • Module contents
    • equiadapt.images package
      • Subpackages
      • Submodules
      • equiadapt.images.utils module
      • Module contents
    • equiadapt.nbody package
      • Subpackages
      • Module contents
    • equiadapt.pointcloud package
      • Subpackages
      • Module contents
  • Module contents
    • BaseCanonicalization
      • BaseCanonicalization.canonicalize()
      • BaseCanonicalization.forward()
      • BaseCanonicalization.invert_canonicalization()
    • 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
      • ContinuousGroupPointcloudCanonicalization.device
      • ContinuousGroupPointcloudCanonicalization.get_groupelement()
      • ContinuousGroupPointcloudCanonicalization.canonicalize()
      • ContinuousGroupPointcloudCanonicalization.canonicalize()
      • ContinuousGroupPointcloudCanonicalization.get_groupelement()
    • ConvNetwork
      • ConvNetwork.__init__()
      • ConvNetwork.forward()
      • ConvNetwork.forward()
    • CustomEquivariantNetwork
      • CustomEquivariantNetwork.__init__()
      • CustomEquivariantNetwork.forward()
      • CustomEquivariantNetwork.forward()
    • 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
      • ESCNNEquivariantNetwork.__init__()
      • ESCNNEquivariantNetwork.forward()
      • ESCNNEquivariantNetwork.forward()
    • ESCNNSteerableNetwork
      • ESCNNSteerableNetwork.__init__()
      • ESCNNSteerableNetwork.forward()
      • ESCNNSteerableNetwork.forward()
    • ESCNNWRNEquivariantNetwork
      • ESCNNWRNEquivariantNetwork.__init__()
      • ESCNNWRNEquivariantNetwork.forward()
      • ESCNNWRNEquivariantNetwork.forward()
    • ESCNNWideBasic
      • ESCNNWideBasic.__init__()
      • ESCNNWideBasic.forward()
      • ESCNNWideBasic.evaluate_output_shape()
      • ESCNNWideBasic.forward()
    • ESCNNWideBottleneck
      • ESCNNWideBottleneck.__init__()
      • ESCNNWideBottleneck.forward()
      • ESCNNWideBottleneck.evaluate_output_shape()
      • ESCNNWideBottleneck.forward()
    • EquivariantPointcloudCanonicalization
      • EquivariantPointcloudCanonicalization.canonicalization_network
      • EquivariantPointcloudCanonicalization.canonicalization_hyperparams
      • EquivariantPointcloudCanonicalization.canonicalization_info_dict
      • EquivariantPointcloudCanonicalization.get_groupelement()
    • GroupEquivariantImageCanonicalization
      • GroupEquivariantImageCanonicalization.__init__()
      • GroupEquivariantImageCanonicalization.get_group_activations()
      • GroupEquivariantImageCanonicalization.get_group_activations()
    • IdentityCanonicalization
      • IdentityCanonicalization.canonicalization_network
      • IdentityCanonicalization.__init__()
      • IdentityCanonicalization.canonicalize()
      • IdentityCanonicalization.canonicalize()
      • IdentityCanonicalization.get_identity_metric()
      • IdentityCanonicalization.get_prior_regularization_loss()
      • IdentityCanonicalization.invert_canonicalization()
    • LieParameterization
      • LieParameterization.group_type
      • LieParameterization.group_dim
      • LieParameterization.get_en_rep()
      • LieParameterization.get_group_rep()
      • LieParameterization.get_on_rep()
      • LieParameterization.get_sen_rep()
      • LieParameterization.get_son_bases()
      • LieParameterization.get_son_rep()
    • 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
      • ResNet18Network.resnet18
      • ResNet18Network.out_vector_size
      • ResNet18Network.forward()
    • RotationEquivariantConv
      • RotationEquivariantConv.__init__()
      • RotationEquivariantConv.get_rotated_permuted_weights()
      • RotationEquivariantConv.forward()
      • RotationEquivariantConv.forward()
      • RotationEquivariantConv.get_rotated_permuted_weights()
    • RotationEquivariantConvLift
      • RotationEquivariantConvLift.__init__()
      • RotationEquivariantConvLift.get_rotated_weights()
      • RotationEquivariantConvLift.forward()
      • RotationEquivariantConvLift.forward()
      • RotationEquivariantConvLift.get_rotated_weights()
    • RotoReflectionEquivariantConv
      • RotoReflectionEquivariantConv.__init__()
      • RotoReflectionEquivariantConv.get_rotoreflected_permuted_weights()
      • RotoReflectionEquivariantConv.forward()
      • RotoReflectionEquivariantConv.forward()
      • RotoReflectionEquivariantConv.get_rotoreflected_permuted_weights()
    • RotoReflectionEquivariantConvLift
      • RotoReflectionEquivariantConvLift.__init__()
      • RotoReflectionEquivariantConvLift.get_rotoreflected_weights()
      • RotoReflectionEquivariantConvLift.forward()
      • RotoReflectionEquivariantConvLift.forward()
      • RotoReflectionEquivariantConvLift.get_rotoreflected_weights()
    • SteerableImageCanonicalization
      • SteerableImageCanonicalization.__init__()
      • SteerableImageCanonicalization.get_rotation_matrix_from_vector()
      • SteerableImageCanonicalization.get_groupelement()
      • SteerableImageCanonicalization.get_groupelement()
      • SteerableImageCanonicalization.get_rotation_matrix_from_vector()
    • VNBatchNorm
      • VNBatchNorm.__init__()
      • VNBatchNorm.forward()
      • VNBatchNorm.forward()
    • VNBilinear
      • VNBilinear.__init__()
      • VNBilinear.forward()
      • VNBilinear.forward()
    • VNLeakyReLU
      • VNLeakyReLU.__init__()
      • VNLeakyReLU.forward()
      • VNLeakyReLU.forward()
    • VNLinear
      • VNLinear.__init__()
      • VNLinear.forward()
      • VNLinear.forward()
    • VNLinearLeakyReLU
      • VNLinearLeakyReLU.__init__()
      • VNLinearLeakyReLU.forward()
      • VNLinearLeakyReLU.forward()
    • VNMaxPool
      • VNMaxPool.__init__()
      • VNMaxPool.forward()
      • VNMaxPool.forward()
    • VNSmall
      • VNSmall.n_knn
      • VNSmall.pooling
      • VNSmall.conv_pos
      • VNSmall.conv1
      • VNSmall.bn1
      • VNSmall.conv2
      • VNSmall.dropout
      • VNSmall.pool
      • VNSmall.__init__()
      • VNSmall.forward()
      • VNSmall.forward()
    • VNSoftplus
      • VNSoftplus.__init__()
      • VNSoftplus.forward()
      • VNSoftplus.forward()
    • VNStdFeature
      • VNStdFeature.dim
      • VNStdFeature.normalize_frame
      • VNStdFeature.__init__()
      • VNStdFeature.forward()
      • VNStdFeature.forward()
    • get_action_on_image_features()
    • get_graph_feature_cross()
    • gram_schmidt()