.coveragerc
.gitignore
.isort.cfg
.pre-commit-config.yaml
.readthedocs.yml
AUTHORS.md
CHANGELOG.md
CONTRIBUTING.md
LICENSE
README.md
conda_env.yaml
min_conda_env.yaml
pyproject.toml
setup.cfg
setup.py
tox.ini
.github/pull_request_template.md
.github/workflows/ci.yml
docs/Makefile
docs/authors.md
docs/changelog.md
docs/conf.py
docs/contributing.md
docs/index.md
docs/license.md
docs/postprocess.py
docs/readme.md
docs/requirements.txt
docs/_static/.gitignore
docs/_static/utils/equiadapt_cat.jpeg
equiadapt/__init__.py
equiadapt/py.typed
equiadapt.egg-info/PKG-INFO
equiadapt.egg-info/SOURCES.txt
equiadapt.egg-info/dependency_links.txt
equiadapt.egg-info/not-zip-safe
equiadapt.egg-info/requires.txt
equiadapt.egg-info/top_level.txt
equiadapt/common/__init__.py
equiadapt/common/basecanonicalization.py
equiadapt/common/utils.py
equiadapt/images/__init__.py
equiadapt/images/utils.py
equiadapt/images/canonicalization/__init__.py
equiadapt/images/canonicalization/continuous_group.py
equiadapt/images/canonicalization/discrete_group.py
equiadapt/images/canonicalization_networks/__init__.py
equiadapt/images/canonicalization_networks/custom_equivariant_networks.py
equiadapt/images/canonicalization_networks/custom_group_equivariant_layers.py
equiadapt/images/canonicalization_networks/custom_nonequivariant_networks.py
equiadapt/images/canonicalization_networks/escnn_networks.py
equiadapt/nbody/__init__.py
equiadapt/nbody/canonicalization/__init__.py
equiadapt/nbody/canonicalization/euclidean_group.py
equiadapt/nbody/canonicalization_networks/__init__.py
equiadapt/nbody/canonicalization_networks/custom_equivariant_networks.py
equiadapt/nbody/canonicalization_networks/custom_group_equivariant_layers.py
equiadapt/pointcloud/__init__.py
equiadapt/pointcloud/canonicalization/__init__.py
equiadapt/pointcloud/canonicalization/continuous_group.py
equiadapt/pointcloud/canonicalization_networks/__init__.py
equiadapt/pointcloud/canonicalization_networks/equivariant_networks.py
equiadapt/pointcloud/canonicalization_networks/vector_neuron_layers.py
examples/__init__.py
examples/images/__init__.py
examples/images/classification/README.md
examples/images/classification/__init__.py
examples/images/classification/inference_utils.py
examples/images/classification/model.py
examples/images/classification/model_utils.py
examples/images/classification/train.py
examples/images/classification/train_utils.py
examples/images/classification/configs/__init__.py
examples/images/classification/configs/default.yaml
examples/images/classification/configs/wandb_sweep.yaml
examples/images/classification/configs/canonicalization/group_equivariant.yaml
examples/images/classification/configs/canonicalization/identity.yaml
examples/images/classification/configs/canonicalization/opt_group_equivariant.yaml
examples/images/classification/configs/canonicalization/opt_steerable.yaml
examples/images/classification/configs/canonicalization/steerable.yaml
examples/images/classification/configs/checkpoint/default.yaml
examples/images/classification/configs/dataset/default.yaml
examples/images/classification/configs/experiment/default.yaml
examples/images/classification/configs/prediction/default.yaml
examples/images/classification/configs/wandb/default.yaml
examples/images/classification/prepare/__init__.py
examples/images/classification/prepare/cifar_data.py
examples/images/classification/prepare/imagenet_data.py
examples/images/classification/prepare/rotated_mnist_data.py
examples/images/classification/prepare/stl10_data.py
examples/images/common/__init__.py
examples/images/common/utils.py
examples/images/segmentation/README.md
examples/images/segmentation/inference_utils.py
examples/images/segmentation/model.py
examples/images/segmentation/model_utils.py
examples/images/segmentation/train.py
examples/images/segmentation/train_utils.py
examples/images/segmentation/configs/__init__.py
examples/images/segmentation/configs/default.yaml
examples/images/segmentation/configs/wandb_sweep.yaml
examples/images/segmentation/configs/canonicalization/group_equivariant.yaml
examples/images/segmentation/configs/canonicalization/identity.yaml
examples/images/segmentation/configs/canonicalization/opt_group_equivariant.yaml
examples/images/segmentation/configs/canonicalization/opt_steerable.yaml
examples/images/segmentation/configs/canonicalization/steerable.yaml
examples/images/segmentation/configs/checkpoint/default.yaml
examples/images/segmentation/configs/dataset/default.yaml
examples/images/segmentation/configs/experiment/default.yaml
examples/images/segmentation/configs/prediction/default.yaml
examples/images/segmentation/prepare/__init__.py
examples/images/segmentation/prepare/coco_data.py
examples/images/segmentation/prepare/vision_transforms.py
examples/nbody/README.md
examples/nbody/model.py
examples/nbody/model_utils.py
examples/nbody/train.py
examples/nbody/train_utils.py
examples/nbody/configs/default.yaml
examples/nbody/configs/canonicalization/vndeepsets.yaml
examples/nbody/configs/checkpoint/default.yaml
examples/nbody/configs/dataset/default.yaml
examples/nbody/configs/experiment/default.yaml
examples/nbody/configs/prediction/default.yaml
examples/nbody/data/n_body_system/._.DS_Store
examples/nbody/data/n_body_system/__init__.py
examples/nbody/data/n_body_system/dataset/__init__.py
examples/nbody/data/n_body_system/dataset/generate_dataset.py
examples/nbody/data/n_body_system/dataset/script.sh
examples/nbody/data/n_body_system/dataset/synthetic_sim.py
examples/nbody/networks/__init__.py
examples/nbody/networks/euclideangraph_base_models.py
examples/nbody/networks/gcl.py
examples/nbody/prepare/nbody_data.py
examples/pointcloud/README.md
examples/pointcloud/__init__.py
examples/pointcloud/classification/README.md
examples/pointcloud/classification/__init__.py
examples/pointcloud/classification/model.py
examples/pointcloud/classification/model_utils.py
examples/pointcloud/classification/prepare.py
examples/pointcloud/classification/train.py
examples/pointcloud/classification/train_utils.py
examples/pointcloud/classification/configs/__init__.py
examples/pointcloud/classification/configs/default.yaml
examples/pointcloud/classification/configs/canonicalization/group_equivariant.yaml
examples/pointcloud/classification/configs/canonicalization/identity.yaml
examples/pointcloud/classification/configs/checkpoint/default.yaml
examples/pointcloud/classification/configs/dataset/default.yaml
examples/pointcloud/classification/configs/experiment/default.yaml
examples/pointcloud/classification/configs/prediction/default.yaml
examples/pointcloud/common/__init__.py
examples/pointcloud/common/networks.py
examples/pointcloud/common/utils.py
examples/pointcloud/part_segmentation/README.md
examples/pointcloud/part_segmentation/__init__.py
examples/pointcloud/part_segmentation/model.py
examples/pointcloud/part_segmentation/model_utils.py
examples/pointcloud/part_segmentation/prepare.py
examples/pointcloud/part_segmentation/train.py
examples/pointcloud/part_segmentation/train_utils.py
examples/pointcloud/part_segmentation/configs/__init__.py
examples/pointcloud/part_segmentation/configs/default.yaml
examples/pointcloud/part_segmentation/configs/canonicalization/group_equivariant.yaml
examples/pointcloud/part_segmentation/configs/canonicalization/identity.yaml
examples/pointcloud/part_segmentation/configs/checkpoint/default.yaml
examples/pointcloud/part_segmentation/configs/dataset/default.yaml
examples/pointcloud/part_segmentation/configs/experiment/default.yaml
examples/pointcloud/part_segmentation/configs/prediction/default.yaml
tests/conftest.py
tests/common/__init__.py
tests/common/test_utils.py
tests/images/__init__.py
tests/images/canonicalization/__init__.py
tests/images/canonicalization/test_continuous_group.py
tests/images/canonicalization/test_discrete_group.py
tutorials/images/classification_group_equivariant_canonicalization.ipynb
tutorials/images/instance_segmentation_group_equivariant_canonicalization.ipynb
tutorials/images/understanding_discrete_canonicalization.ipynb
tutorials/images/utils/equivariant_wrn_c4.pt
tutorials/images/utils/sample_data.pt
tutorials/nbody/nbody.ipynb
utils/equiadapt_cat.jpeg
utils/equiadapt_logo.png
utils/equiadapt_sam.gif
utils/logo.png