Index

_

__init__() (equiadapt.common.basecanonicalization.ContinuousGroupCanonicalization method) (equiadapt.common.basecanonicalization.DiscreteGroupCanonicalization method) (equiadapt.common.basecanonicalization.IdentityCanonicalization method) (equiadapt.common.ContinuousGroupCanonicalization method) (equiadapt.common.DiscreteGroupCanonicalization method) (equiadapt.common.IdentityCanonicalization method) (equiadapt.ContinuousGroupCanonicalization method) (equiadapt.ContinuousGroupImageCanonicalization method) (equiadapt.ConvNetwork method) (equiadapt.CustomEquivariantNetwork method) (equiadapt.DiscreteGroupCanonicalization method) (equiadapt.DiscreteGroupImageCanonicalization method) (equiadapt.ESCNNEquivariantNetwork method) (equiadapt.ESCNNSteerableNetwork method) (equiadapt.ESCNNWideBasic method) (equiadapt.ESCNNWideBottleneck method) (equiadapt.ESCNNWRNEquivariantNetwork method) (equiadapt.GroupEquivariantImageCanonicalization method) (equiadapt.IdentityCanonicalization method) (equiadapt.images.canonicalization.continuous_group.ContinuousGroupImageCanonicalization method) (equiadapt.images.canonicalization.continuous_group.OptimizedSteerableImageCanonicalization method) (equiadapt.images.canonicalization.continuous_group.SteerableImageCanonicalization method) (equiadapt.images.canonicalization.ContinuousGroupImageCanonicalization method) (equiadapt.images.canonicalization.discrete_group.DiscreteGroupImageCanonicalization method) (equiadapt.images.canonicalization.discrete_group.GroupEquivariantImageCanonicalization method) (equiadapt.images.canonicalization.discrete_group.OptimizedGroupEquivariantImageCanonicalization method) (equiadapt.images.canonicalization.DiscreteGroupImageCanonicalization method) (equiadapt.images.canonicalization.GroupEquivariantImageCanonicalization method) (equiadapt.images.canonicalization.OptimizedGroupEquivariantImageCanonicalization method) (equiadapt.images.canonicalization.OptimizedSteerableImageCanonicalization method) (equiadapt.images.canonicalization.SteerableImageCanonicalization method) (equiadapt.images.canonicalization_networks.ConvNetwork method) (equiadapt.images.canonicalization_networks.custom_equivariant_networks.CustomEquivariantNetwork method) (equiadapt.images.canonicalization_networks.custom_group_equivariant_layers.RotationEquivariantConv method) (equiadapt.images.canonicalization_networks.custom_group_equivariant_layers.RotationEquivariantConvLift method) (equiadapt.images.canonicalization_networks.custom_group_equivariant_layers.RotoReflectionEquivariantConv method) (equiadapt.images.canonicalization_networks.custom_group_equivariant_layers.RotoReflectionEquivariantConvLift method) (equiadapt.images.canonicalization_networks.custom_nonequivariant_networks.ConvNetwork method) (equiadapt.images.canonicalization_networks.CustomEquivariantNetwork method) (equiadapt.images.canonicalization_networks.escnn_networks.ESCNNEquivariantNetwork method) (equiadapt.images.canonicalization_networks.escnn_networks.ESCNNSteerableNetwork method) (equiadapt.images.canonicalization_networks.escnn_networks.ESCNNWideBasic method) (equiadapt.images.canonicalization_networks.escnn_networks.ESCNNWideBottleneck method) (equiadapt.images.canonicalization_networks.escnn_networks.ESCNNWRNEquivariantNetwork method) (equiadapt.images.canonicalization_networks.ESCNNEquivariantNetwork method) (equiadapt.images.canonicalization_networks.ESCNNSteerableNetwork method) (equiadapt.images.canonicalization_networks.ESCNNWideBasic method) (equiadapt.images.canonicalization_networks.ESCNNWideBottleneck method) (equiadapt.images.canonicalization_networks.ESCNNWRNEquivariantNetwork method) (equiadapt.images.canonicalization_networks.RotationEquivariantConv method) (equiadapt.images.canonicalization_networks.RotationEquivariantConvLift method) (equiadapt.images.canonicalization_networks.RotoReflectionEquivariantConv method) (equiadapt.images.canonicalization_networks.RotoReflectionEquivariantConvLift method) (equiadapt.images.ContinuousGroupImageCanonicalization method) (equiadapt.images.ConvNetwork method) (equiadapt.images.CustomEquivariantNetwork method) (equiadapt.images.DiscreteGroupImageCanonicalization method) (equiadapt.images.ESCNNEquivariantNetwork method) (equiadapt.images.ESCNNSteerableNetwork method) (equiadapt.images.ESCNNWideBasic method) (equiadapt.images.ESCNNWideBottleneck method) (equiadapt.images.ESCNNWRNEquivariantNetwork method) (equiadapt.images.GroupEquivariantImageCanonicalization method) (equiadapt.images.OptimizedGroupEquivariantImageCanonicalization method) (equiadapt.images.OptimizedSteerableImageCanonicalization method) (equiadapt.images.RotationEquivariantConv method) (equiadapt.images.RotationEquivariantConvLift method) (equiadapt.images.RotoReflectionEquivariantConv method) (equiadapt.images.RotoReflectionEquivariantConvLift method) (equiadapt.images.SteerableImageCanonicalization method) (equiadapt.OptimizedGroupEquivariantImageCanonicalization method) (equiadapt.OptimizedSteerableImageCanonicalization method) (equiadapt.pointcloud.canonicalization_networks.equivariant_networks.VNSmall method) (equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers.VNBatchNorm method) (equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers.VNBilinear method) (equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers.VNLeakyReLU method) (equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers.VNLinear method) (equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers.VNLinearLeakyReLU method) (equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers.VNMaxPool method) (equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers.VNSoftplus method) (equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers.VNStdFeature method) (equiadapt.pointcloud.canonicalization_networks.VNBatchNorm method) (equiadapt.pointcloud.canonicalization_networks.VNBilinear method) (equiadapt.pointcloud.canonicalization_networks.VNLeakyReLU method) (equiadapt.pointcloud.canonicalization_networks.VNLinear method) (equiadapt.pointcloud.canonicalization_networks.VNLinearLeakyReLU method) (equiadapt.pointcloud.canonicalization_networks.VNMaxPool method) (equiadapt.pointcloud.canonicalization_networks.VNSmall method) (equiadapt.pointcloud.canonicalization_networks.VNSoftplus method) (equiadapt.pointcloud.canonicalization_networks.VNStdFeature method) (equiadapt.pointcloud.VNBatchNorm method) (equiadapt.pointcloud.VNBilinear method) (equiadapt.pointcloud.VNLeakyReLU method) (equiadapt.pointcloud.VNLinear method) (equiadapt.pointcloud.VNLinearLeakyReLU method) (equiadapt.pointcloud.VNMaxPool method) (equiadapt.pointcloud.VNSmall method) (equiadapt.pointcloud.VNSoftplus method) (equiadapt.pointcloud.VNStdFeature method) (equiadapt.RotationEquivariantConv method) (equiadapt.RotationEquivariantConvLift method) (equiadapt.RotoReflectionEquivariantConv method) (equiadapt.RotoReflectionEquivariantConvLift method) (equiadapt.SteerableImageCanonicalization method) (equiadapt.VNBatchNorm method) (equiadapt.VNBilinear method) (equiadapt.VNLeakyReLU method) (equiadapt.VNLinear method) (equiadapt.VNLinearLeakyReLU method) (equiadapt.VNMaxPool method) (equiadapt.VNSmall method) (equiadapt.VNSoftplus method) (equiadapt.VNStdFeature method)

B

BaseCanonicalization (class in equiadapt) (class in equiadapt.common) (class in equiadapt.common.basecanonicalization) beta (equiadapt.common.basecanonicalization.ContinuousGroupCanonicalization attribute) (equiadapt.common.basecanonicalization.DiscreteGroupCanonicalization attribute) (equiadapt.common.ContinuousGroupCanonicalization attribute) (equiadapt.common.DiscreteGroupCanonicalization attribute) (equiadapt.ContinuousGroupCanonicalization attribute) (equiadapt.DiscreteGroupCanonicalization attribute)

bn1 (equiadapt.pointcloud.canonicalization_networks.equivariant_networks.VNSmall attribute) (equiadapt.pointcloud.canonicalization_networks.VNSmall attribute) (equiadapt.pointcloud.VNSmall attribute) (equiadapt.VNSmall attribute)

C

canonicalization_hyperparams (equiadapt.EquivariantPointcloudCanonicalization attribute) (equiadapt.pointcloud.canonicalization.continuous_group.EquivariantPointcloudCanonicalization attribute) (equiadapt.pointcloud.canonicalization.EquivariantPointcloudCanonicalization attribute) (equiadapt.pointcloud.EquivariantPointcloudCanonicalization attribute) canonicalization_info_dict (equiadapt.EquivariantPointcloudCanonicalization attribute) (equiadapt.nbody.canonicalization.euclidean_group.EuclideanGroupNBody attribute) (equiadapt.pointcloud.canonicalization.continuous_group.EquivariantPointcloudCanonicalization attribute) (equiadapt.pointcloud.canonicalization.EquivariantPointcloudCanonicalization attribute) (equiadapt.pointcloud.EquivariantPointcloudCanonicalization attribute) canonicalization_network (equiadapt.common.basecanonicalization.ContinuousGroupCanonicalization attribute) (equiadapt.common.basecanonicalization.DiscreteGroupCanonicalization attribute) (equiadapt.common.basecanonicalization.IdentityCanonicalization attribute) (equiadapt.common.ContinuousGroupCanonicalization attribute) (equiadapt.common.DiscreteGroupCanonicalization attribute) (equiadapt.common.IdentityCanonicalization attribute) (equiadapt.ContinuousGroupCanonicalization attribute) (equiadapt.DiscreteGroupCanonicalization attribute) (equiadapt.EquivariantPointcloudCanonicalization attribute) (equiadapt.IdentityCanonicalization attribute) (equiadapt.pointcloud.canonicalization.continuous_group.EquivariantPointcloudCanonicalization attribute) (equiadapt.pointcloud.canonicalization.EquivariantPointcloudCanonicalization attribute) (equiadapt.pointcloud.EquivariantPointcloudCanonicalization attribute) canonicalizationnetworkout_to_groupelement() (equiadapt.common.basecanonicalization.ContinuousGroupCanonicalization method), [1] (equiadapt.common.ContinuousGroupCanonicalization method), [1] (equiadapt.ContinuousGroupCanonicalization method), [1] canonicalize() (equiadapt.BaseCanonicalization method) (equiadapt.common.BaseCanonicalization method) (equiadapt.common.basecanonicalization.BaseCanonicalization method) (equiadapt.common.basecanonicalization.ContinuousGroupCanonicalization method), [1] (equiadapt.common.basecanonicalization.DiscreteGroupCanonicalization method), [1] (equiadapt.common.basecanonicalization.IdentityCanonicalization method), [1] (equiadapt.common.ContinuousGroupCanonicalization method), [1] (equiadapt.common.DiscreteGroupCanonicalization method), [1] (equiadapt.common.IdentityCanonicalization method), [1] (equiadapt.ContinuousGroupCanonicalization method), [1] (equiadapt.ContinuousGroupImageCanonicalization method), [1] (equiadapt.ContinuousGroupPointcloudCanonicalization method), [1] (equiadapt.DiscreteGroupCanonicalization method), [1] (equiadapt.DiscreteGroupImageCanonicalization method), [1] (equiadapt.IdentityCanonicalization method), [1] (equiadapt.images.canonicalization.continuous_group.ContinuousGroupImageCanonicalization method), [1] (equiadapt.images.canonicalization.ContinuousGroupImageCanonicalization method), [1] (equiadapt.images.canonicalization.discrete_group.DiscreteGroupImageCanonicalization method), [1] (equiadapt.images.canonicalization.DiscreteGroupImageCanonicalization method), [1] (equiadapt.images.ContinuousGroupImageCanonicalization method), [1] (equiadapt.images.DiscreteGroupImageCanonicalization method), [1] (equiadapt.nbody.canonicalization.euclidean_group.EuclideanGroupNBody method) (equiadapt.pointcloud.canonicalization.continuous_group.ContinuousGroupPointcloudCanonicalization method), [1] (equiadapt.pointcloud.canonicalization.ContinuousGroupPointcloudCanonicalization method), [1] (equiadapt.pointcloud.ContinuousGroupPointcloudCanonicalization method), [1]

ContinuousGroupCanonicalization (class in equiadapt) (class in equiadapt.common) (class in equiadapt.common.basecanonicalization) ContinuousGroupImageCanonicalization (class in equiadapt) (class in equiadapt.images) (class in equiadapt.images.canonicalization) (class in equiadapt.images.canonicalization.continuous_group) ContinuousGroupPointcloudCanonicalization (class in equiadapt) (class in equiadapt.pointcloud) (class in equiadapt.pointcloud.canonicalization) (class in equiadapt.pointcloud.canonicalization.continuous_group) conv1 (equiadapt.pointcloud.canonicalization_networks.equivariant_networks.VNSmall attribute) (equiadapt.pointcloud.canonicalization_networks.VNSmall attribute) (equiadapt.pointcloud.VNSmall attribute) (equiadapt.VNSmall attribute) conv2 (equiadapt.pointcloud.canonicalization_networks.equivariant_networks.VNSmall attribute) (equiadapt.pointcloud.canonicalization_networks.VNSmall attribute) (equiadapt.pointcloud.VNSmall attribute) (equiadapt.VNSmall attribute) conv_pos (equiadapt.pointcloud.canonicalization_networks.equivariant_networks.VNSmall attribute) (equiadapt.pointcloud.canonicalization_networks.VNSmall attribute) (equiadapt.pointcloud.VNSmall attribute) (equiadapt.VNSmall attribute) ConvNetwork (class in equiadapt) (class in equiadapt.images) (class in equiadapt.images.canonicalization_networks) (class in equiadapt.images.canonicalization_networks.custom_nonequivariant_networks) CustomEquivariantNetwork (class in equiadapt) (class in equiadapt.images) (class in equiadapt.images.canonicalization_networks) (class in equiadapt.images.canonicalization_networks.custom_equivariant_networks)

D

device (equiadapt.ContinuousGroupPointcloudCanonicalization attribute) (equiadapt.pointcloud.canonicalization.continuous_group.ContinuousGroupPointcloudCanonicalization attribute) (equiadapt.pointcloud.canonicalization.ContinuousGroupPointcloudCanonicalization attribute) (equiadapt.pointcloud.ContinuousGroupPointcloudCanonicalization attribute) dim (equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers.VNStdFeature attribute) (equiadapt.pointcloud.canonicalization_networks.VNStdFeature attribute) (equiadapt.pointcloud.VNStdFeature attribute) (equiadapt.VNStdFeature attribute) DiscreteGroupCanonicalization (class in equiadapt) (class in equiadapt.common) (class in equiadapt.common.basecanonicalization)

DiscreteGroupImageCanonicalization (class in equiadapt) (class in equiadapt.images) (class in equiadapt.images.canonicalization) (class in equiadapt.images.canonicalization.discrete_group) dropout (equiadapt.pointcloud.canonicalization_networks.equivariant_networks.VNSmall attribute) (equiadapt.pointcloud.canonicalization_networks.VNSmall attribute) (equiadapt.pointcloud.VNSmall attribute) (equiadapt.VNSmall attribute)

E

equiadapt module equiadapt.common module equiadapt.common.basecanonicalization module equiadapt.common.utils module equiadapt.images module equiadapt.images.canonicalization module equiadapt.images.canonicalization.continuous_group module equiadapt.images.canonicalization.discrete_group module equiadapt.images.canonicalization_networks module equiadapt.images.canonicalization_networks.custom_equivariant_networks module equiadapt.images.canonicalization_networks.custom_group_equivariant_layers module equiadapt.images.canonicalization_networks.custom_nonequivariant_networks module equiadapt.images.canonicalization_networks.escnn_networks module equiadapt.images.utils module equiadapt.nbody module equiadapt.nbody.canonicalization module equiadapt.nbody.canonicalization.euclidean_group module equiadapt.nbody.canonicalization_networks module equiadapt.nbody.canonicalization_networks.custom_group_equivariant_layers module equiadapt.pointcloud module equiadapt.pointcloud.canonicalization module

equiadapt.pointcloud.canonicalization.continuous_group module equiadapt.pointcloud.canonicalization_networks module equiadapt.pointcloud.canonicalization_networks.equivariant_networks module equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers module EquivariantPointcloudCanonicalization (class in equiadapt) (class in equiadapt.pointcloud) (class in equiadapt.pointcloud.canonicalization) (class in equiadapt.pointcloud.canonicalization.continuous_group) ESCNNEquivariantNetwork (class in equiadapt) (class in equiadapt.images) (class in equiadapt.images.canonicalization_networks) (class in equiadapt.images.canonicalization_networks.escnn_networks) ESCNNSteerableNetwork (class in equiadapt) (class in equiadapt.images) (class in equiadapt.images.canonicalization_networks) (class in equiadapt.images.canonicalization_networks.escnn_networks) ESCNNWideBasic (class in equiadapt) (class in equiadapt.images) (class in equiadapt.images.canonicalization_networks) (class in equiadapt.images.canonicalization_networks.escnn_networks) ESCNNWideBottleneck (class in equiadapt) (class in equiadapt.images) (class in equiadapt.images.canonicalization_networks) (class in equiadapt.images.canonicalization_networks.escnn_networks) ESCNNWRNEquivariantNetwork (class in equiadapt) (class in equiadapt.images) (class in equiadapt.images.canonicalization_networks) (class in equiadapt.images.canonicalization_networks.escnn_networks) EuclideanGroupNBody (class in equiadapt.nbody.canonicalization.euclidean_group) evaluate_output_shape() (equiadapt.ESCNNWideBasic method) (equiadapt.ESCNNWideBottleneck method) (equiadapt.images.canonicalization_networks.escnn_networks.ESCNNWideBasic method) (equiadapt.images.canonicalization_networks.escnn_networks.ESCNNWideBottleneck method) (equiadapt.images.canonicalization_networks.ESCNNWideBasic method) (equiadapt.images.canonicalization_networks.ESCNNWideBottleneck method) (equiadapt.images.ESCNNWideBasic method) (equiadapt.images.ESCNNWideBottleneck method)

F

flip_boxes() (in module equiadapt.images) (in module equiadapt.images.utils) flip_masks() (in module equiadapt.images) (in module equiadapt.images.utils) forward() (equiadapt.BaseCanonicalization method) (equiadapt.common.BaseCanonicalization method) (equiadapt.common.basecanonicalization.BaseCanonicalization method) (equiadapt.ConvNetwork method), [1] (equiadapt.CustomEquivariantNetwork method), [1] (equiadapt.ESCNNEquivariantNetwork method), [1] (equiadapt.ESCNNSteerableNetwork method), [1] (equiadapt.ESCNNWideBasic method), [1] (equiadapt.ESCNNWideBottleneck method), [1] (equiadapt.ESCNNWRNEquivariantNetwork method), [1] (equiadapt.images.canonicalization_networks.ConvNetwork method), [1] (equiadapt.images.canonicalization_networks.custom_equivariant_networks.CustomEquivariantNetwork method), [1] (equiadapt.images.canonicalization_networks.custom_group_equivariant_layers.RotationEquivariantConv method), [1] (equiadapt.images.canonicalization_networks.custom_group_equivariant_layers.RotationEquivariantConvLift method), [1] (equiadapt.images.canonicalization_networks.custom_group_equivariant_layers.RotoReflectionEquivariantConv method), [1] (equiadapt.images.canonicalization_networks.custom_group_equivariant_layers.RotoReflectionEquivariantConvLift method), [1] (equiadapt.images.canonicalization_networks.custom_nonequivariant_networks.ConvNetwork method), [1] (equiadapt.images.canonicalization_networks.custom_nonequivariant_networks.ResNet18Network method) (equiadapt.images.canonicalization_networks.CustomEquivariantNetwork method), [1] (equiadapt.images.canonicalization_networks.escnn_networks.ESCNNEquivariantNetwork method), [1] (equiadapt.images.canonicalization_networks.escnn_networks.ESCNNSteerableNetwork method), [1] (equiadapt.images.canonicalization_networks.escnn_networks.ESCNNWideBasic method), [1] (equiadapt.images.canonicalization_networks.escnn_networks.ESCNNWideBottleneck method), [1] (equiadapt.images.canonicalization_networks.escnn_networks.ESCNNWRNEquivariantNetwork method), [1] (equiadapt.images.canonicalization_networks.ESCNNEquivariantNetwork method), [1] (equiadapt.images.canonicalization_networks.ESCNNSteerableNetwork method), [1] (equiadapt.images.canonicalization_networks.ESCNNWideBasic method), [1] (equiadapt.images.canonicalization_networks.ESCNNWideBottleneck method), [1] (equiadapt.images.canonicalization_networks.ESCNNWRNEquivariantNetwork method), [1] (equiadapt.images.canonicalization_networks.ResNet18Network method) (equiadapt.images.canonicalization_networks.RotationEquivariantConv method), [1] (equiadapt.images.canonicalization_networks.RotationEquivariantConvLift method), [1] (equiadapt.images.canonicalization_networks.RotoReflectionEquivariantConv method), [1] (equiadapt.images.canonicalization_networks.RotoReflectionEquivariantConvLift method), [1] (equiadapt.images.ConvNetwork method), [1] (equiadapt.images.CustomEquivariantNetwork method), [1] (equiadapt.images.ESCNNEquivariantNetwork method), [1] (equiadapt.images.ESCNNSteerableNetwork method), [1] (equiadapt.images.ESCNNWideBasic method), [1] (equiadapt.images.ESCNNWideBottleneck method), [1] (equiadapt.images.ESCNNWRNEquivariantNetwork method), [1] (equiadapt.images.ResNet18Network method) (equiadapt.images.RotationEquivariantConv method), [1] (equiadapt.images.RotationEquivariantConvLift method), [1] (equiadapt.images.RotoReflectionEquivariantConv method), [1] (equiadapt.images.RotoReflectionEquivariantConvLift method), [1] (equiadapt.nbody.canonicalization.euclidean_group.EuclideanGroupNBody method) (equiadapt.nbody.canonicalization_networks.custom_group_equivariant_layers.VNLeakyReLU method) (equiadapt.nbody.canonicalization_networks.custom_group_equivariant_layers.VNSoftplus method) (equiadapt.pointcloud.canonicalization_networks.equivariant_networks.VNSmall method), [1] (equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers.VNBatchNorm method), [1] (equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers.VNBilinear method), [1] (equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers.VNLeakyReLU method), [1] (equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers.VNLinear method), [1] (equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers.VNLinearLeakyReLU method), [1] (equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers.VNMaxPool method), [1] (equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers.VNSoftplus method), [1] (equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers.VNStdFeature method), [1] (equiadapt.pointcloud.canonicalization_networks.VNBatchNorm method), [1] (equiadapt.pointcloud.canonicalization_networks.VNBilinear method), [1] (equiadapt.pointcloud.canonicalization_networks.VNLeakyReLU method), [1] (equiadapt.pointcloud.canonicalization_networks.VNLinear method), [1] (equiadapt.pointcloud.canonicalization_networks.VNLinearLeakyReLU method), [1] (equiadapt.pointcloud.canonicalization_networks.VNMaxPool method), [1] (equiadapt.pointcloud.canonicalization_networks.VNSmall method), [1] (equiadapt.pointcloud.canonicalization_networks.VNSoftplus method), [1] (equiadapt.pointcloud.canonicalization_networks.VNStdFeature method), [1] (equiadapt.pointcloud.VNBatchNorm method), [1] (equiadapt.pointcloud.VNBilinear method), [1] (equiadapt.pointcloud.VNLeakyReLU method), [1] (equiadapt.pointcloud.VNLinear method), [1] (equiadapt.pointcloud.VNLinearLeakyReLU method), [1] (equiadapt.pointcloud.VNMaxPool method), [1] (equiadapt.pointcloud.VNSmall method), [1] (equiadapt.pointcloud.VNSoftplus method), [1] (equiadapt.pointcloud.VNStdFeature method), [1] (equiadapt.ResNet18Network method) (equiadapt.RotationEquivariantConv method), [1] (equiadapt.RotationEquivariantConvLift method), [1] (equiadapt.RotoReflectionEquivariantConv method), [1] (equiadapt.RotoReflectionEquivariantConvLift method), [1] (equiadapt.VNBatchNorm method), [1] (equiadapt.VNBilinear method), [1] (equiadapt.VNLeakyReLU method), [1] (equiadapt.VNLinear method), [1] (equiadapt.VNLinearLeakyReLU method), [1] (equiadapt.VNMaxPool method), [1] (equiadapt.VNSmall method), [1] (equiadapt.VNSoftplus method), [1] (equiadapt.VNStdFeature method), [1]

G

get_action_on_image_features() (in module equiadapt) (in module equiadapt.images) (in module equiadapt.images.utils) get_en_rep() (equiadapt.common.LieParameterization method) (equiadapt.common.utils.LieParameterization method) (equiadapt.LieParameterization method) get_graph_feature_cross() (in module equiadapt) (in module equiadapt.pointcloud) (in module equiadapt.pointcloud.canonicalization_networks) (in module equiadapt.pointcloud.canonicalization_networks.equivariant_networks) get_group_activations() (equiadapt.DiscreteGroupImageCanonicalization method) (equiadapt.GroupEquivariantImageCanonicalization method), [1] (equiadapt.images.canonicalization.discrete_group.DiscreteGroupImageCanonicalization method) (equiadapt.images.canonicalization.discrete_group.GroupEquivariantImageCanonicalization method), [1] (equiadapt.images.canonicalization.discrete_group.OptimizedGroupEquivariantImageCanonicalization method), [1] (equiadapt.images.canonicalization.DiscreteGroupImageCanonicalization method) (equiadapt.images.canonicalization.GroupEquivariantImageCanonicalization method), [1] (equiadapt.images.canonicalization.OptimizedGroupEquivariantImageCanonicalization method), [1] (equiadapt.images.DiscreteGroupImageCanonicalization method) (equiadapt.images.GroupEquivariantImageCanonicalization method), [1] (equiadapt.images.OptimizedGroupEquivariantImageCanonicalization method), [1] (equiadapt.OptimizedGroupEquivariantImageCanonicalization method), [1] get_group_from_out_vectors() (equiadapt.ContinuousGroupImageCanonicalization method), [1] (equiadapt.images.canonicalization.continuous_group.ContinuousGroupImageCanonicalization method), [1] (equiadapt.images.canonicalization.ContinuousGroupImageCanonicalization method), [1] (equiadapt.images.ContinuousGroupImageCanonicalization method), [1] get_group_rep() (equiadapt.common.LieParameterization method) (equiadapt.common.utils.LieParameterization method) (equiadapt.LieParameterization method) get_groupelement() (equiadapt.ContinuousGroupImageCanonicalization method), [1] (equiadapt.ContinuousGroupPointcloudCanonicalization method), [1] (equiadapt.DiscreteGroupImageCanonicalization method), [1] (equiadapt.EquivariantPointcloudCanonicalization method) (equiadapt.images.canonicalization.continuous_group.ContinuousGroupImageCanonicalization method), [1] (equiadapt.images.canonicalization.continuous_group.OptimizedSteerableImageCanonicalization method), [1] (equiadapt.images.canonicalization.continuous_group.SteerableImageCanonicalization method), [1] (equiadapt.images.canonicalization.ContinuousGroupImageCanonicalization method), [1] (equiadapt.images.canonicalization.discrete_group.DiscreteGroupImageCanonicalization method), [1] (equiadapt.images.canonicalization.DiscreteGroupImageCanonicalization method), [1] (equiadapt.images.canonicalization.OptimizedSteerableImageCanonicalization method), [1] (equiadapt.images.canonicalization.SteerableImageCanonicalization method), [1] (equiadapt.images.ContinuousGroupImageCanonicalization method), [1] (equiadapt.images.DiscreteGroupImageCanonicalization method), [1] (equiadapt.images.OptimizedSteerableImageCanonicalization method), [1] (equiadapt.images.SteerableImageCanonicalization method), [1] (equiadapt.nbody.canonicalization.euclidean_group.EuclideanGroupNBody method) (equiadapt.OptimizedSteerableImageCanonicalization method), [1] (equiadapt.pointcloud.canonicalization.continuous_group.ContinuousGroupPointcloudCanonicalization method), [1] (equiadapt.pointcloud.canonicalization.continuous_group.EquivariantPointcloudCanonicalization method) (equiadapt.pointcloud.canonicalization.ContinuousGroupPointcloudCanonicalization method), [1] (equiadapt.pointcloud.canonicalization.EquivariantPointcloudCanonicalization method) (equiadapt.pointcloud.ContinuousGroupPointcloudCanonicalization method), [1] (equiadapt.pointcloud.EquivariantPointcloudCanonicalization method) (equiadapt.SteerableImageCanonicalization method), [1] get_identity_metric() (equiadapt.common.basecanonicalization.ContinuousGroupCanonicalization method), [1] (equiadapt.common.basecanonicalization.DiscreteGroupCanonicalization method), [1] (equiadapt.common.basecanonicalization.IdentityCanonicalization method) (equiadapt.common.ContinuousGroupCanonicalization method), [1] (equiadapt.common.DiscreteGroupCanonicalization method), [1] (equiadapt.common.IdentityCanonicalization method) (equiadapt.ContinuousGroupCanonicalization method), [1] (equiadapt.DiscreteGroupCanonicalization method), [1] (equiadapt.IdentityCanonicalization method) get_on_rep() (equiadapt.common.LieParameterization method) (equiadapt.common.utils.LieParameterization method) (equiadapt.LieParameterization method) get_optimization_specific_loss() (equiadapt.images.canonicalization.continuous_group.OptimizedSteerableImageCanonicalization method), [1] (equiadapt.images.canonicalization.discrete_group.OptimizedGroupEquivariantImageCanonicalization method), [1] (equiadapt.images.canonicalization.OptimizedGroupEquivariantImageCanonicalization method), [1] (equiadapt.images.canonicalization.OptimizedSteerableImageCanonicalization method), [1] (equiadapt.images.OptimizedGroupEquivariantImageCanonicalization method), [1] (equiadapt.images.OptimizedSteerableImageCanonicalization method), [1] (equiadapt.OptimizedGroupEquivariantImageCanonicalization method), [1] (equiadapt.OptimizedSteerableImageCanonicalization method), [1] get_prior_regularization_loss() (equiadapt.common.basecanonicalization.ContinuousGroupCanonicalization method), [1] (equiadapt.common.basecanonicalization.DiscreteGroupCanonicalization method), [1] (equiadapt.common.basecanonicalization.IdentityCanonicalization method) (equiadapt.common.ContinuousGroupCanonicalization method), [1] (equiadapt.common.DiscreteGroupCanonicalization method), [1] (equiadapt.common.IdentityCanonicalization method) (equiadapt.ContinuousGroupCanonicalization method), [1] (equiadapt.DiscreteGroupCanonicalization method), [1] (equiadapt.IdentityCanonicalization method)

get_rotated_permuted_weights() (equiadapt.images.canonicalization_networks.custom_group_equivariant_layers.RotationEquivariantConv method), [1] (equiadapt.images.canonicalization_networks.RotationEquivariantConv method), [1] (equiadapt.images.RotationEquivariantConv method), [1] (equiadapt.RotationEquivariantConv method), [1] get_rotated_weights() (equiadapt.images.canonicalization_networks.custom_group_equivariant_layers.RotationEquivariantConvLift method), [1] (equiadapt.images.canonicalization_networks.RotationEquivariantConvLift method), [1] (equiadapt.images.RotationEquivariantConvLift method), [1] (equiadapt.RotationEquivariantConvLift method), [1] get_rotation_matrix_from_vector() (equiadapt.ContinuousGroupImageCanonicalization method) (equiadapt.images.canonicalization.continuous_group.ContinuousGroupImageCanonicalization method) (equiadapt.images.canonicalization.continuous_group.OptimizedSteerableImageCanonicalization method), [1] (equiadapt.images.canonicalization.continuous_group.SteerableImageCanonicalization method), [1] (equiadapt.images.canonicalization.ContinuousGroupImageCanonicalization method) (equiadapt.images.canonicalization.OptimizedSteerableImageCanonicalization method), [1] (equiadapt.images.canonicalization.SteerableImageCanonicalization method), [1] (equiadapt.images.ContinuousGroupImageCanonicalization method) (equiadapt.images.OptimizedSteerableImageCanonicalization method), [1] (equiadapt.images.SteerableImageCanonicalization method), [1] (equiadapt.OptimizedSteerableImageCanonicalization method), [1] (equiadapt.SteerableImageCanonicalization method), [1] get_rotoreflected_permuted_weights() (equiadapt.images.canonicalization_networks.custom_group_equivariant_layers.RotoReflectionEquivariantConv method), [1] (equiadapt.images.canonicalization_networks.RotoReflectionEquivariantConv method), [1] (equiadapt.images.RotoReflectionEquivariantConv method), [1] (equiadapt.RotoReflectionEquivariantConv method), [1] get_rotoreflected_weights() (equiadapt.images.canonicalization_networks.custom_group_equivariant_layers.RotoReflectionEquivariantConvLift method), [1] (equiadapt.images.canonicalization_networks.RotoReflectionEquivariantConvLift method), [1] (equiadapt.images.RotoReflectionEquivariantConvLift method), [1] (equiadapt.RotoReflectionEquivariantConvLift method), [1] get_sen_rep() (equiadapt.common.LieParameterization method) (equiadapt.common.utils.LieParameterization method) (equiadapt.LieParameterization method) get_son_bases() (equiadapt.common.LieParameterization method) (equiadapt.common.utils.LieParameterization method) (equiadapt.LieParameterization method) get_son_rep() (equiadapt.common.LieParameterization method) (equiadapt.common.utils.LieParameterization method) (equiadapt.LieParameterization method) gradient_trick (equiadapt.common.basecanonicalization.DiscreteGroupCanonicalization attribute) (equiadapt.common.DiscreteGroupCanonicalization attribute) (equiadapt.DiscreteGroupCanonicalization attribute) gram_schmidt() (in module equiadapt) (in module equiadapt.common) (in module equiadapt.common.utils) group_augment() (equiadapt.images.canonicalization.continuous_group.OptimizedSteerableImageCanonicalization method), [1] (equiadapt.images.canonicalization.discrete_group.OptimizedGroupEquivariantImageCanonicalization method), [1] (equiadapt.images.canonicalization.OptimizedGroupEquivariantImageCanonicalization method), [1] (equiadapt.images.canonicalization.OptimizedSteerableImageCanonicalization method), [1] (equiadapt.images.OptimizedGroupEquivariantImageCanonicalization method), [1] (equiadapt.images.OptimizedSteerableImageCanonicalization method), [1] (equiadapt.OptimizedGroupEquivariantImageCanonicalization method), [1] (equiadapt.OptimizedSteerableImageCanonicalization method), [1] group_dim (equiadapt.common.LieParameterization attribute) (equiadapt.common.utils.LieParameterization attribute) (equiadapt.LieParameterization attribute) group_type (equiadapt.common.LieParameterization attribute) (equiadapt.common.utils.LieParameterization attribute) (equiadapt.LieParameterization attribute) groupactivations_to_groupelement() (equiadapt.DiscreteGroupImageCanonicalization method), [1] (equiadapt.images.canonicalization.discrete_group.DiscreteGroupImageCanonicalization method), [1] (equiadapt.images.canonicalization.DiscreteGroupImageCanonicalization method), [1] (equiadapt.images.DiscreteGroupImageCanonicalization method), [1] groupactivations_to_groupelementonehot() (equiadapt.common.basecanonicalization.DiscreteGroupCanonicalization method), [1] (equiadapt.common.DiscreteGroupCanonicalization method), [1] (equiadapt.DiscreteGroupCanonicalization method), [1] GroupEquivariantImageCanonicalization (class in equiadapt) (class in equiadapt.images) (class in equiadapt.images.canonicalization) (class in equiadapt.images.canonicalization.discrete_group)

I

IdentityCanonicalization (class in equiadapt) (class in equiadapt.common) (class in equiadapt.common.basecanonicalization) invert_canonicalization() (equiadapt.BaseCanonicalization method) (equiadapt.common.BaseCanonicalization method) (equiadapt.common.basecanonicalization.BaseCanonicalization method) (equiadapt.common.basecanonicalization.ContinuousGroupCanonicalization method), [1] (equiadapt.common.basecanonicalization.DiscreteGroupCanonicalization method), [1] (equiadapt.common.basecanonicalization.IdentityCanonicalization method) (equiadapt.common.ContinuousGroupCanonicalization method), [1] (equiadapt.common.DiscreteGroupCanonicalization method), [1] (equiadapt.common.IdentityCanonicalization method) (equiadapt.ContinuousGroupCanonicalization method), [1] (equiadapt.ContinuousGroupImageCanonicalization method), [1] (equiadapt.DiscreteGroupCanonicalization method), [1] (equiadapt.DiscreteGroupImageCanonicalization method), [1] (equiadapt.IdentityCanonicalization method) (equiadapt.images.canonicalization.continuous_group.ContinuousGroupImageCanonicalization method), [1] (equiadapt.images.canonicalization.ContinuousGroupImageCanonicalization method), [1] (equiadapt.images.canonicalization.discrete_group.DiscreteGroupImageCanonicalization method), [1] (equiadapt.images.canonicalization.DiscreteGroupImageCanonicalization method), [1] (equiadapt.images.ContinuousGroupImageCanonicalization method), [1] (equiadapt.images.DiscreteGroupImageCanonicalization method), [1] (equiadapt.nbody.canonicalization.euclidean_group.EuclideanGroupNBody method)

K

knn() (in module equiadapt.pointcloud) (in module equiadapt.pointcloud.canonicalization_networks) (in module equiadapt.pointcloud.canonicalization_networks.equivariant_networks)

L

LieParameterization (class in equiadapt) (class in equiadapt.common) (class in equiadapt.common.utils)

M

mean_pool() (in module equiadapt.pointcloud) (in module equiadapt.pointcloud.canonicalization_networks) (in module equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers) modified_gram_schmidt() (equiadapt.nbody.canonicalization.euclidean_group.EuclideanGroupNBody method) module equiadapt equiadapt.common equiadapt.common.basecanonicalization equiadapt.common.utils equiadapt.images equiadapt.images.canonicalization equiadapt.images.canonicalization.continuous_group equiadapt.images.canonicalization.discrete_group equiadapt.images.canonicalization_networks equiadapt.images.canonicalization_networks.custom_equivariant_networks equiadapt.images.canonicalization_networks.custom_group_equivariant_layers equiadapt.images.canonicalization_networks.custom_nonequivariant_networks equiadapt.images.canonicalization_networks.escnn_networks equiadapt.images.utils equiadapt.nbody equiadapt.nbody.canonicalization equiadapt.nbody.canonicalization.euclidean_group equiadapt.nbody.canonicalization_networks equiadapt.nbody.canonicalization_networks.custom_group_equivariant_layers equiadapt.pointcloud equiadapt.pointcloud.canonicalization equiadapt.pointcloud.canonicalization.continuous_group equiadapt.pointcloud.canonicalization_networks equiadapt.pointcloud.canonicalization_networks.equivariant_networks equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers

N

n_knn (equiadapt.pointcloud.canonicalization_networks.equivariant_networks.VNSmall attribute) (equiadapt.pointcloud.canonicalization_networks.VNSmall attribute) (equiadapt.pointcloud.VNSmall attribute) (equiadapt.VNSmall attribute)

normalize_frame (equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers.VNStdFeature attribute) (equiadapt.pointcloud.canonicalization_networks.VNStdFeature attribute) (equiadapt.pointcloud.VNStdFeature attribute) (equiadapt.VNStdFeature attribute)

O

OptimizedGroupEquivariantImageCanonicalization (class in equiadapt) (class in equiadapt.images) (class in equiadapt.images.canonicalization) (class in equiadapt.images.canonicalization.discrete_group) OptimizedSteerableImageCanonicalization (class in equiadapt) (class in equiadapt.images) (class in equiadapt.images.canonicalization) (class in equiadapt.images.canonicalization.continuous_group)

out_vector_size (equiadapt.images.canonicalization_networks.custom_nonequivariant_networks.ResNet18Network attribute) (equiadapt.images.canonicalization_networks.ResNet18Network attribute) (equiadapt.images.ResNet18Network attribute) (equiadapt.ResNet18Network attribute)

P

pool (equiadapt.pointcloud.canonicalization_networks.equivariant_networks.VNSmall attribute) (equiadapt.pointcloud.canonicalization_networks.VNSmall attribute) (equiadapt.pointcloud.VNSmall attribute) (equiadapt.VNSmall attribute)

pooling (equiadapt.pointcloud.canonicalization_networks.equivariant_networks.VNSmall attribute) (equiadapt.pointcloud.canonicalization_networks.VNSmall attribute) (equiadapt.pointcloud.VNSmall attribute) (equiadapt.VNSmall attribute)

R

resnet18 (equiadapt.images.canonicalization_networks.custom_nonequivariant_networks.ResNet18Network attribute) (equiadapt.images.canonicalization_networks.ResNet18Network attribute) (equiadapt.images.ResNet18Network attribute) (equiadapt.ResNet18Network attribute) ResNet18Network (class in equiadapt) (class in equiadapt.images) (class in equiadapt.images.canonicalization_networks) (class in equiadapt.images.canonicalization_networks.custom_nonequivariant_networks) roll_by_gather() (in module equiadapt.images) (in module equiadapt.images.utils) rotate_and_maybe_reflect() (equiadapt.images.canonicalization.discrete_group.OptimizedGroupEquivariantImageCanonicalization method), [1] (equiadapt.images.canonicalization.OptimizedGroupEquivariantImageCanonicalization method), [1] (equiadapt.images.OptimizedGroupEquivariantImageCanonicalization method), [1] (equiadapt.OptimizedGroupEquivariantImageCanonicalization method), [1] rotate_boxes() (in module equiadapt.images) (in module equiadapt.images.utils) rotate_masks() (in module equiadapt.images) (in module equiadapt.images.utils)

rotate_points() (in module equiadapt.images) (in module equiadapt.images.utils) RotationEquivariantConv (class in equiadapt) (class in equiadapt.images) (class in equiadapt.images.canonicalization_networks) (class in equiadapt.images.canonicalization_networks.custom_group_equivariant_layers) RotationEquivariantConvLift (class in equiadapt) (class in equiadapt.images) (class in equiadapt.images.canonicalization_networks) (class in equiadapt.images.canonicalization_networks.custom_group_equivariant_layers) RotoReflectionEquivariantConv (class in equiadapt) (class in equiadapt.images) (class in equiadapt.images.canonicalization_networks) (class in equiadapt.images.canonicalization_networks.custom_group_equivariant_layers) RotoReflectionEquivariantConvLift (class in equiadapt) (class in equiadapt.images) (class in equiadapt.images.canonicalization_networks) (class in equiadapt.images.canonicalization_networks.custom_group_equivariant_layers)

S

SteerableImageCanonicalization (class in equiadapt) (class in equiadapt.images) (class in equiadapt.images.canonicalization) (class in equiadapt.images.canonicalization.continuous_group)

T

transformations_before_canonicalization_network_forward() (equiadapt.ContinuousGroupImageCanonicalization method), [1] (equiadapt.DiscreteGroupImageCanonicalization method), [1] (equiadapt.images.canonicalization.continuous_group.ContinuousGroupImageCanonicalization method), [1] (equiadapt.images.canonicalization.ContinuousGroupImageCanonicalization method), [1] (equiadapt.images.canonicalization.discrete_group.DiscreteGroupImageCanonicalization method), [1] (equiadapt.images.canonicalization.DiscreteGroupImageCanonicalization method), [1] (equiadapt.images.ContinuousGroupImageCanonicalization method), [1] (equiadapt.images.DiscreteGroupImageCanonicalization method), [1]

V

VNBatchNorm (class in equiadapt) (class in equiadapt.pointcloud) (class in equiadapt.pointcloud.canonicalization_networks) (class in equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers) VNBilinear (class in equiadapt) (class in equiadapt.pointcloud) (class in equiadapt.pointcloud.canonicalization_networks) (class in equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers) VNLeakyReLU (class in equiadapt) (class in equiadapt.nbody.canonicalization_networks.custom_group_equivariant_layers) (class in equiadapt.pointcloud) (class in equiadapt.pointcloud.canonicalization_networks) (class in equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers) VNLinear (class in equiadapt) (class in equiadapt.pointcloud) (class in equiadapt.pointcloud.canonicalization_networks) (class in equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers) VNLinearLeakyReLU (class in equiadapt) (class in equiadapt.pointcloud) (class in equiadapt.pointcloud.canonicalization_networks) (class in equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers)

VNMaxPool (class in equiadapt) (class in equiadapt.pointcloud) (class in equiadapt.pointcloud.canonicalization_networks) (class in equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers) VNSmall (class in equiadapt) (class in equiadapt.pointcloud) (class in equiadapt.pointcloud.canonicalization_networks) (class in equiadapt.pointcloud.canonicalization_networks.equivariant_networks) VNSoftplus (class in equiadapt) (class in equiadapt.nbody.canonicalization_networks.custom_group_equivariant_layers) (class in equiadapt.pointcloud) (class in equiadapt.pointcloud.canonicalization_networks) (class in equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers) VNStdFeature (class in equiadapt) (class in equiadapt.pointcloud) (class in equiadapt.pointcloud.canonicalization_networks) (class in equiadapt.pointcloud.canonicalization_networks.vector_neuron_layers)