├── .gitignore ├── LICENSE ├── README.md ├── plenoxels ├── __init__.py ├── configs │ ├── __init__.py │ ├── config_helpers.py │ └── final │ │ ├── D-NeRF │ │ ├── dnerf_explicit.py │ │ └── dnerf_hybrid.py │ │ ├── DyNeRF │ │ ├── dynerf_explicit.py │ │ └── dynerf_hybrid.py │ │ ├── LLFF │ │ ├── llff_explicit.py │ │ └── llff_hybrid.py │ │ ├── NeRF │ │ ├── nerf_explicit.py │ │ └── nerf_hybrid.py │ │ └── Phototourism │ │ ├── brandenburg_explicit.py │ │ ├── brandenburg_hybrid.py │ │ ├── sacrecoeur_explicit.py │ │ ├── sacrecoeur_hybrid.py │ │ ├── trevi_explicit.py │ │ └── trevi_hybrid.py ├── datasets │ ├── __init__.py │ ├── base_dataset.py │ ├── colmap_utils.py │ ├── data_loading.py │ ├── intrinsics.py │ ├── llff_dataset.py │ ├── phototourism_dataset.py │ ├── ray_utils.py │ ├── synthetic_nerf_dataset.py │ └── video_datasets.py ├── main.py ├── models │ ├── __init__.py │ ├── density_fields.py │ ├── kplane_field.py │ └── lowrank_model.py ├── ops │ ├── __init__.py │ ├── activations.py │ ├── flip │ │ ├── __init__.py │ │ ├── data.py │ │ ├── flip.py │ │ └── flip_api.py │ ├── image │ │ ├── __init__.py │ │ ├── io.py │ │ └── metrics.py │ ├── interpolation.py │ ├── losses │ │ ├── __init__.py │ │ ├── distortion_loss.py │ │ └── histogram_loss.py │ └── lr_scheduling.py ├── raymarching │ ├── __init__.py │ ├── ray_samplers.py │ └── spatial_distortions.py ├── runners │ ├── __init__.py │ ├── base_trainer.py │ ├── phototourism_trainer.py │ ├── regularization.py │ ├── static_trainer.py │ └── video_trainer.py └── utils │ ├── __init__.py │ ├── create_rendering.py │ ├── ema.py │ ├── eval_model_metrics.py │ ├── my_tqdm.py │ ├── parse_args.py │ └── timer.py └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/README.md -------------------------------------------------------------------------------- /plenoxels/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /plenoxels/configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/configs/__init__.py -------------------------------------------------------------------------------- /plenoxels/configs/config_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/configs/config_helpers.py -------------------------------------------------------------------------------- /plenoxels/configs/final/D-NeRF/dnerf_explicit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/configs/final/D-NeRF/dnerf_explicit.py -------------------------------------------------------------------------------- /plenoxels/configs/final/D-NeRF/dnerf_hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/configs/final/D-NeRF/dnerf_hybrid.py -------------------------------------------------------------------------------- /plenoxels/configs/final/DyNeRF/dynerf_explicit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/configs/final/DyNeRF/dynerf_explicit.py -------------------------------------------------------------------------------- /plenoxels/configs/final/DyNeRF/dynerf_hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/configs/final/DyNeRF/dynerf_hybrid.py -------------------------------------------------------------------------------- /plenoxels/configs/final/LLFF/llff_explicit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/configs/final/LLFF/llff_explicit.py -------------------------------------------------------------------------------- /plenoxels/configs/final/LLFF/llff_hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/configs/final/LLFF/llff_hybrid.py -------------------------------------------------------------------------------- /plenoxels/configs/final/NeRF/nerf_explicit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/configs/final/NeRF/nerf_explicit.py -------------------------------------------------------------------------------- /plenoxels/configs/final/NeRF/nerf_hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/configs/final/NeRF/nerf_hybrid.py -------------------------------------------------------------------------------- /plenoxels/configs/final/Phototourism/brandenburg_explicit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/configs/final/Phototourism/brandenburg_explicit.py -------------------------------------------------------------------------------- /plenoxels/configs/final/Phototourism/brandenburg_hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/configs/final/Phototourism/brandenburg_hybrid.py -------------------------------------------------------------------------------- /plenoxels/configs/final/Phototourism/sacrecoeur_explicit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/configs/final/Phototourism/sacrecoeur_explicit.py -------------------------------------------------------------------------------- /plenoxels/configs/final/Phototourism/sacrecoeur_hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/configs/final/Phototourism/sacrecoeur_hybrid.py -------------------------------------------------------------------------------- /plenoxels/configs/final/Phototourism/trevi_explicit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/configs/final/Phototourism/trevi_explicit.py -------------------------------------------------------------------------------- /plenoxels/configs/final/Phototourism/trevi_hybrid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/configs/final/Phototourism/trevi_hybrid.py -------------------------------------------------------------------------------- /plenoxels/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/datasets/__init__.py -------------------------------------------------------------------------------- /plenoxels/datasets/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/datasets/base_dataset.py -------------------------------------------------------------------------------- /plenoxels/datasets/colmap_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/datasets/colmap_utils.py -------------------------------------------------------------------------------- /plenoxels/datasets/data_loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/datasets/data_loading.py -------------------------------------------------------------------------------- /plenoxels/datasets/intrinsics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/datasets/intrinsics.py -------------------------------------------------------------------------------- /plenoxels/datasets/llff_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/datasets/llff_dataset.py -------------------------------------------------------------------------------- /plenoxels/datasets/phototourism_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/datasets/phototourism_dataset.py -------------------------------------------------------------------------------- /plenoxels/datasets/ray_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/datasets/ray_utils.py -------------------------------------------------------------------------------- /plenoxels/datasets/synthetic_nerf_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/datasets/synthetic_nerf_dataset.py -------------------------------------------------------------------------------- /plenoxels/datasets/video_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/datasets/video_datasets.py -------------------------------------------------------------------------------- /plenoxels/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/main.py -------------------------------------------------------------------------------- /plenoxels/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /plenoxels/models/density_fields.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/models/density_fields.py -------------------------------------------------------------------------------- /plenoxels/models/kplane_field.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/models/kplane_field.py -------------------------------------------------------------------------------- /plenoxels/models/lowrank_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/models/lowrank_model.py -------------------------------------------------------------------------------- /plenoxels/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /plenoxels/ops/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/ops/activations.py -------------------------------------------------------------------------------- /plenoxels/ops/flip/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /plenoxels/ops/flip/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/ops/flip/data.py -------------------------------------------------------------------------------- /plenoxels/ops/flip/flip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/ops/flip/flip.py -------------------------------------------------------------------------------- /plenoxels/ops/flip/flip_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/ops/flip/flip_api.py -------------------------------------------------------------------------------- /plenoxels/ops/image/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /plenoxels/ops/image/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/ops/image/io.py -------------------------------------------------------------------------------- /plenoxels/ops/image/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/ops/image/metrics.py -------------------------------------------------------------------------------- /plenoxels/ops/interpolation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/ops/interpolation.py -------------------------------------------------------------------------------- /plenoxels/ops/losses/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /plenoxels/ops/losses/distortion_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/ops/losses/distortion_loss.py -------------------------------------------------------------------------------- /plenoxels/ops/losses/histogram_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/ops/losses/histogram_loss.py -------------------------------------------------------------------------------- /plenoxels/ops/lr_scheduling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/ops/lr_scheduling.py -------------------------------------------------------------------------------- /plenoxels/raymarching/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /plenoxels/raymarching/ray_samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/raymarching/ray_samplers.py -------------------------------------------------------------------------------- /plenoxels/raymarching/spatial_distortions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/raymarching/spatial_distortions.py -------------------------------------------------------------------------------- /plenoxels/runners/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /plenoxels/runners/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/runners/base_trainer.py -------------------------------------------------------------------------------- /plenoxels/runners/phototourism_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/runners/phototourism_trainer.py -------------------------------------------------------------------------------- /plenoxels/runners/regularization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/runners/regularization.py -------------------------------------------------------------------------------- /plenoxels/runners/static_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/runners/static_trainer.py -------------------------------------------------------------------------------- /plenoxels/runners/video_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/runners/video_trainer.py -------------------------------------------------------------------------------- /plenoxels/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /plenoxels/utils/create_rendering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/utils/create_rendering.py -------------------------------------------------------------------------------- /plenoxels/utils/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/utils/ema.py -------------------------------------------------------------------------------- /plenoxels/utils/eval_model_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/utils/eval_model_metrics.py -------------------------------------------------------------------------------- /plenoxels/utils/my_tqdm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/utils/my_tqdm.py -------------------------------------------------------------------------------- /plenoxels/utils/parse_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/utils/parse_args.py -------------------------------------------------------------------------------- /plenoxels/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/plenoxels/utils/timer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sarafridov/K-Planes/HEAD/requirements.txt --------------------------------------------------------------------------------