├── .gitignore ├── LICENSE ├── README.md ├── configs ├── compnode │ └── 1gpu.yaml ├── datamanager │ ├── camera │ │ └── off.yaml │ ├── mp3d_gZ6f7yhEvPG.yaml │ ├── mp3d_pLe4wQe7qrG.yaml │ ├── replica.yaml │ ├── replica_room0.yaml │ └── replica_room2.yaml ├── model │ ├── blueprint_nerfacto.yaml │ ├── blueprint_siren.yaml │ └── stereo.yaml ├── mp3d_gZ6f7yhEvPG.yaml ├── mp3d_pLe4wQe7qrG.yaml ├── optim │ ├── blueprint_nerfacto.yaml │ ├── field │ │ ├── adam_nerfacto.yaml │ │ └── adam_siren.yaml │ ├── nerfacto.yaml │ ├── proposal │ │ └── adam.yaml │ └── stereo.yaml ├── replica_room0.yaml ├── replica_room2.yaml └── trainer │ └── vanilla.yaml ├── requirements.txt ├── src ├── callbacks │ └── metrics.py ├── data │ ├── datamanagers │ │ ├── blueprint_datamanager.py │ │ └── semantic_datamanager.py │ ├── dataparsers │ │ └── replica_dataparser.py │ └── datasets │ │ ├── rgbsem_dataset.py │ │ └── semantic_dataset.py ├── engine │ └── trainer.py ├── eval.py ├── fields │ ├── blunf_field.py │ ├── nerfacto_field.py │ ├── populate_fields.py │ ├── siren_field.py │ ├── spatial_nerfacto.py │ └── spatial_vanillanerf.py ├── infer.py ├── init │ ├── init_datamanager.py │ ├── init_model.py │ ├── init_optimizer.py │ └── init_trainer.py ├── model_components │ ├── extra_renderer.py │ └── point_generator.py ├── models │ ├── blueprint_nerfacto.py │ ├── nerfacto.py │ └── stereo.py └── train.py └── utils ├── camera_utils.py ├── colormap.py ├── file_utils.py └── viz_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/README.md -------------------------------------------------------------------------------- /configs/compnode/1gpu.yaml: -------------------------------------------------------------------------------- 1 | device: cuda 2 | world_size: 1 3 | local_rank: 0 -------------------------------------------------------------------------------- /configs/datamanager/camera/off.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/configs/datamanager/camera/off.yaml -------------------------------------------------------------------------------- /configs/datamanager/mp3d_gZ6f7yhEvPG.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/configs/datamanager/mp3d_gZ6f7yhEvPG.yaml -------------------------------------------------------------------------------- /configs/datamanager/mp3d_pLe4wQe7qrG.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/configs/datamanager/mp3d_pLe4wQe7qrG.yaml -------------------------------------------------------------------------------- /configs/datamanager/replica.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/configs/datamanager/replica.yaml -------------------------------------------------------------------------------- /configs/datamanager/replica_room0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/configs/datamanager/replica_room0.yaml -------------------------------------------------------------------------------- /configs/datamanager/replica_room2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/configs/datamanager/replica_room2.yaml -------------------------------------------------------------------------------- /configs/model/blueprint_nerfacto.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/configs/model/blueprint_nerfacto.yaml -------------------------------------------------------------------------------- /configs/model/blueprint_siren.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/configs/model/blueprint_siren.yaml -------------------------------------------------------------------------------- /configs/model/stereo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/configs/model/stereo.yaml -------------------------------------------------------------------------------- /configs/mp3d_gZ6f7yhEvPG.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/configs/mp3d_gZ6f7yhEvPG.yaml -------------------------------------------------------------------------------- /configs/mp3d_pLe4wQe7qrG.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/configs/mp3d_pLe4wQe7qrG.yaml -------------------------------------------------------------------------------- /configs/optim/blueprint_nerfacto.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/configs/optim/blueprint_nerfacto.yaml -------------------------------------------------------------------------------- /configs/optim/field/adam_nerfacto.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/configs/optim/field/adam_nerfacto.yaml -------------------------------------------------------------------------------- /configs/optim/field/adam_siren.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/configs/optim/field/adam_siren.yaml -------------------------------------------------------------------------------- /configs/optim/nerfacto.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/configs/optim/nerfacto.yaml -------------------------------------------------------------------------------- /configs/optim/proposal/adam.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/configs/optim/proposal/adam.yaml -------------------------------------------------------------------------------- /configs/optim/stereo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/configs/optim/stereo.yaml -------------------------------------------------------------------------------- /configs/replica_room0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/configs/replica_room0.yaml -------------------------------------------------------------------------------- /configs/replica_room2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/configs/replica_room2.yaml -------------------------------------------------------------------------------- /configs/trainer/vanilla.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/configs/trainer/vanilla.yaml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/callbacks/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/callbacks/metrics.py -------------------------------------------------------------------------------- /src/data/datamanagers/blueprint_datamanager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/data/datamanagers/blueprint_datamanager.py -------------------------------------------------------------------------------- /src/data/datamanagers/semantic_datamanager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/data/datamanagers/semantic_datamanager.py -------------------------------------------------------------------------------- /src/data/dataparsers/replica_dataparser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/data/dataparsers/replica_dataparser.py -------------------------------------------------------------------------------- /src/data/datasets/rgbsem_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/data/datasets/rgbsem_dataset.py -------------------------------------------------------------------------------- /src/data/datasets/semantic_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/data/datasets/semantic_dataset.py -------------------------------------------------------------------------------- /src/engine/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/engine/trainer.py -------------------------------------------------------------------------------- /src/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/eval.py -------------------------------------------------------------------------------- /src/fields/blunf_field.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/fields/blunf_field.py -------------------------------------------------------------------------------- /src/fields/nerfacto_field.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/fields/nerfacto_field.py -------------------------------------------------------------------------------- /src/fields/populate_fields.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/fields/populate_fields.py -------------------------------------------------------------------------------- /src/fields/siren_field.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/fields/siren_field.py -------------------------------------------------------------------------------- /src/fields/spatial_nerfacto.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/fields/spatial_nerfacto.py -------------------------------------------------------------------------------- /src/fields/spatial_vanillanerf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/fields/spatial_vanillanerf.py -------------------------------------------------------------------------------- /src/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/infer.py -------------------------------------------------------------------------------- /src/init/init_datamanager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/init/init_datamanager.py -------------------------------------------------------------------------------- /src/init/init_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/init/init_model.py -------------------------------------------------------------------------------- /src/init/init_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/init/init_optimizer.py -------------------------------------------------------------------------------- /src/init/init_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/init/init_trainer.py -------------------------------------------------------------------------------- /src/model_components/extra_renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/model_components/extra_renderer.py -------------------------------------------------------------------------------- /src/model_components/point_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/model_components/point_generator.py -------------------------------------------------------------------------------- /src/models/blueprint_nerfacto.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/models/blueprint_nerfacto.py -------------------------------------------------------------------------------- /src/models/nerfacto.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/models/nerfacto.py -------------------------------------------------------------------------------- /src/models/stereo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/models/stereo.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/src/train.py -------------------------------------------------------------------------------- /utils/camera_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/utils/camera_utils.py -------------------------------------------------------------------------------- /utils/colormap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/utils/colormap.py -------------------------------------------------------------------------------- /utils/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/utils/file_utils.py -------------------------------------------------------------------------------- /utils/viz_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robincourant/blunf/HEAD/utils/viz_utils.py --------------------------------------------------------------------------------