├── .gitignore ├── .vscode ├── launch.json └── settings.json ├── LICENSE ├── README.md ├── install_tutel.md ├── requirements.txt └── switch_nerf ├── __init__.py ├── configs └── switch_nerf │ ├── building.yaml │ ├── bungee.yaml │ ├── campus.yaml │ ├── mission_bay.yaml │ ├── residence.yaml │ ├── rubble.yaml │ └── sci-art.yaml ├── datasets ├── __init__.py ├── block_filesystem_dataset.py ├── dataset_utils.py ├── filesystem_dataset.py ├── lists │ ├── block_nerf_id_map.json │ ├── block_nerf_train.txt │ ├── block_nerf_train_val.txt │ ├── block_nerf_train_val_dummy.txt │ └── block_nerf_val.txt ├── memory_dataset.py └── nerf_data │ ├── __init__.py │ ├── load_LINEMOD.py │ ├── load_blender.py │ ├── load_bungee.py │ ├── load_deepvoxels.py │ ├── load_gigapixel.py │ ├── load_llff.py │ ├── nerf_loader.py │ └── ray_utils.py ├── eval.py ├── eval_ckpt.py ├── eval_image.py ├── eval_image_blocknerf.py ├── eval_nerf_moe.py ├── eval_points.py ├── image_metadata.py ├── metrics.py ├── misc_utils.py ├── models ├── __init__.py ├── cascade.py ├── mega_nerf.py ├── mega_nerf_container.py ├── model_utils.py ├── nerf.py └── nerf_moe.py ├── modules ├── __init__.py └── tutel_moe_ext │ ├── torch_moe_layer_nobatch.py │ ├── tutel_communicate_nobatch.py │ ├── tutel_fast_dispatch.py │ ├── tutel_fast_dispatch_nobatch.py │ ├── tutel_moe_layer_nobatch.py │ ├── tutel_moe_nobatch.py │ ├── tutel_sparse_nobatch.py │ └── tutel_system.py ├── opts.py ├── opts_nerf.py ├── ray_utils.py ├── rendering.py ├── rendering_mip.py ├── runner.py ├── scripts ├── convert_to_container_moe.py ├── copy_images.py ├── create_octree_moe.py └── merge_points.py ├── spherical_harmonics.py ├── train.py ├── train_nerf_moe.py └── utils ├── __init__.py ├── functions.py └── logger.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/.gitignore -------------------------------------------------------------------------------- /.vscode/launch.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/.vscode/launch.json -------------------------------------------------------------------------------- /.vscode/settings.json: -------------------------------------------------------------------------------- 1 | { 2 | } -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/README.md -------------------------------------------------------------------------------- /install_tutel.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/install_tutel.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/requirements.txt -------------------------------------------------------------------------------- /switch_nerf/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /switch_nerf/configs/switch_nerf/building.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/configs/switch_nerf/building.yaml -------------------------------------------------------------------------------- /switch_nerf/configs/switch_nerf/bungee.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/configs/switch_nerf/bungee.yaml -------------------------------------------------------------------------------- /switch_nerf/configs/switch_nerf/campus.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/configs/switch_nerf/campus.yaml -------------------------------------------------------------------------------- /switch_nerf/configs/switch_nerf/mission_bay.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/configs/switch_nerf/mission_bay.yaml -------------------------------------------------------------------------------- /switch_nerf/configs/switch_nerf/residence.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/configs/switch_nerf/residence.yaml -------------------------------------------------------------------------------- /switch_nerf/configs/switch_nerf/rubble.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/configs/switch_nerf/rubble.yaml -------------------------------------------------------------------------------- /switch_nerf/configs/switch_nerf/sci-art.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/configs/switch_nerf/sci-art.yaml -------------------------------------------------------------------------------- /switch_nerf/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /switch_nerf/datasets/block_filesystem_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/datasets/block_filesystem_dataset.py -------------------------------------------------------------------------------- /switch_nerf/datasets/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/datasets/dataset_utils.py -------------------------------------------------------------------------------- /switch_nerf/datasets/filesystem_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/datasets/filesystem_dataset.py -------------------------------------------------------------------------------- /switch_nerf/datasets/lists/block_nerf_id_map.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/datasets/lists/block_nerf_id_map.json -------------------------------------------------------------------------------- /switch_nerf/datasets/lists/block_nerf_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/datasets/lists/block_nerf_train.txt -------------------------------------------------------------------------------- /switch_nerf/datasets/lists/block_nerf_train_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/datasets/lists/block_nerf_train_val.txt -------------------------------------------------------------------------------- /switch_nerf/datasets/lists/block_nerf_train_val_dummy.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/datasets/lists/block_nerf_train_val_dummy.txt -------------------------------------------------------------------------------- /switch_nerf/datasets/lists/block_nerf_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/datasets/lists/block_nerf_val.txt -------------------------------------------------------------------------------- /switch_nerf/datasets/memory_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/datasets/memory_dataset.py -------------------------------------------------------------------------------- /switch_nerf/datasets/nerf_data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /switch_nerf/datasets/nerf_data/load_LINEMOD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/datasets/nerf_data/load_LINEMOD.py -------------------------------------------------------------------------------- /switch_nerf/datasets/nerf_data/load_blender.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/datasets/nerf_data/load_blender.py -------------------------------------------------------------------------------- /switch_nerf/datasets/nerf_data/load_bungee.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/datasets/nerf_data/load_bungee.py -------------------------------------------------------------------------------- /switch_nerf/datasets/nerf_data/load_deepvoxels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/datasets/nerf_data/load_deepvoxels.py -------------------------------------------------------------------------------- /switch_nerf/datasets/nerf_data/load_gigapixel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/datasets/nerf_data/load_gigapixel.py -------------------------------------------------------------------------------- /switch_nerf/datasets/nerf_data/load_llff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/datasets/nerf_data/load_llff.py -------------------------------------------------------------------------------- /switch_nerf/datasets/nerf_data/nerf_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/datasets/nerf_data/nerf_loader.py -------------------------------------------------------------------------------- /switch_nerf/datasets/nerf_data/ray_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/datasets/nerf_data/ray_utils.py -------------------------------------------------------------------------------- /switch_nerf/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/eval.py -------------------------------------------------------------------------------- /switch_nerf/eval_ckpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/eval_ckpt.py -------------------------------------------------------------------------------- /switch_nerf/eval_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/eval_image.py -------------------------------------------------------------------------------- /switch_nerf/eval_image_blocknerf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/eval_image_blocknerf.py -------------------------------------------------------------------------------- /switch_nerf/eval_nerf_moe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/eval_nerf_moe.py -------------------------------------------------------------------------------- /switch_nerf/eval_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/eval_points.py -------------------------------------------------------------------------------- /switch_nerf/image_metadata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/image_metadata.py -------------------------------------------------------------------------------- /switch_nerf/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/metrics.py -------------------------------------------------------------------------------- /switch_nerf/misc_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/misc_utils.py -------------------------------------------------------------------------------- /switch_nerf/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /switch_nerf/models/cascade.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/models/cascade.py -------------------------------------------------------------------------------- /switch_nerf/models/mega_nerf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/models/mega_nerf.py -------------------------------------------------------------------------------- /switch_nerf/models/mega_nerf_container.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/models/mega_nerf_container.py -------------------------------------------------------------------------------- /switch_nerf/models/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/models/model_utils.py -------------------------------------------------------------------------------- /switch_nerf/models/nerf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/models/nerf.py -------------------------------------------------------------------------------- /switch_nerf/models/nerf_moe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/models/nerf_moe.py -------------------------------------------------------------------------------- /switch_nerf/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /switch_nerf/modules/tutel_moe_ext/torch_moe_layer_nobatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/modules/tutel_moe_ext/torch_moe_layer_nobatch.py -------------------------------------------------------------------------------- /switch_nerf/modules/tutel_moe_ext/tutel_communicate_nobatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/modules/tutel_moe_ext/tutel_communicate_nobatch.py -------------------------------------------------------------------------------- /switch_nerf/modules/tutel_moe_ext/tutel_fast_dispatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/modules/tutel_moe_ext/tutel_fast_dispatch.py -------------------------------------------------------------------------------- /switch_nerf/modules/tutel_moe_ext/tutel_fast_dispatch_nobatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/modules/tutel_moe_ext/tutel_fast_dispatch_nobatch.py -------------------------------------------------------------------------------- /switch_nerf/modules/tutel_moe_ext/tutel_moe_layer_nobatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/modules/tutel_moe_ext/tutel_moe_layer_nobatch.py -------------------------------------------------------------------------------- /switch_nerf/modules/tutel_moe_ext/tutel_moe_nobatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/modules/tutel_moe_ext/tutel_moe_nobatch.py -------------------------------------------------------------------------------- /switch_nerf/modules/tutel_moe_ext/tutel_sparse_nobatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/modules/tutel_moe_ext/tutel_sparse_nobatch.py -------------------------------------------------------------------------------- /switch_nerf/modules/tutel_moe_ext/tutel_system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/modules/tutel_moe_ext/tutel_system.py -------------------------------------------------------------------------------- /switch_nerf/opts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/opts.py -------------------------------------------------------------------------------- /switch_nerf/opts_nerf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/opts_nerf.py -------------------------------------------------------------------------------- /switch_nerf/ray_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/ray_utils.py -------------------------------------------------------------------------------- /switch_nerf/rendering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/rendering.py -------------------------------------------------------------------------------- /switch_nerf/rendering_mip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/rendering_mip.py -------------------------------------------------------------------------------- /switch_nerf/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/runner.py -------------------------------------------------------------------------------- /switch_nerf/scripts/convert_to_container_moe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/scripts/convert_to_container_moe.py -------------------------------------------------------------------------------- /switch_nerf/scripts/copy_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/scripts/copy_images.py -------------------------------------------------------------------------------- /switch_nerf/scripts/create_octree_moe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/scripts/create_octree_moe.py -------------------------------------------------------------------------------- /switch_nerf/scripts/merge_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/scripts/merge_points.py -------------------------------------------------------------------------------- /switch_nerf/spherical_harmonics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/spherical_harmonics.py -------------------------------------------------------------------------------- /switch_nerf/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/train.py -------------------------------------------------------------------------------- /switch_nerf/train_nerf_moe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/train_nerf_moe.py -------------------------------------------------------------------------------- /switch_nerf/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /switch_nerf/utils/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/utils/functions.py -------------------------------------------------------------------------------- /switch_nerf/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MiZhenxing/Switch-NeRF/HEAD/switch_nerf/utils/logger.py --------------------------------------------------------------------------------