├── .gitignore ├── LICENSE ├── README.md ├── configs ├── render_config.gin └── train_neuropump.gin ├── doc └── teaser.png ├── img_process ├── estimate.py └── lighten_crop.py ├── internal ├── camera_utils.py ├── configs.py ├── coord.py ├── datasets.py ├── geopoly.py ├── image.py ├── math.py ├── models.py ├── pycolmap │ ├── .gitignore │ ├── LICENSE.txt │ ├── README.md │ ├── pycolmap │ │ ├── __init__.py │ │ ├── camera.py │ │ ├── database.py │ │ ├── image.py │ │ ├── rotation.py │ │ └── scene_manager.py │ └── tools │ │ ├── colmap_to_nvm.py │ │ ├── delete_images.py │ │ ├── impute_missing_cameras.py │ │ ├── save_cameras_as_ply.py │ │ ├── transform_model.py │ │ ├── write_camera_track_to_bundler.py │ │ └── write_depthmap_to_ply.py ├── raw_utils.py ├── ref_utils.py ├── render.py ├── stepfun.py ├── train_utils.py ├── utils.py └── vis.py ├── render.py ├── requirements.txt ├── scripts ├── render_neuropump.sh ├── render_neuropump_extension_change_background_light.sh ├── render_neuropump_extension_change_refraction_index.sh ├── render_neuropump_extension_change_s.sh └── train_neuropump.sh ├── tests ├── camera_utils_test.py ├── coord_test.py ├── datasets_test.py ├── geopoly_test.py ├── image_test.py ├── math_test.py ├── ref_utils_test.py ├── render_test.py ├── stepfun_test.py └── utils_test.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/README.md -------------------------------------------------------------------------------- /configs/render_config.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/configs/render_config.gin -------------------------------------------------------------------------------- /configs/train_neuropump.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/configs/train_neuropump.gin -------------------------------------------------------------------------------- /doc/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/doc/teaser.png -------------------------------------------------------------------------------- /img_process/estimate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/img_process/estimate.py -------------------------------------------------------------------------------- /img_process/lighten_crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/img_process/lighten_crop.py -------------------------------------------------------------------------------- /internal/camera_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/camera_utils.py -------------------------------------------------------------------------------- /internal/configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/configs.py -------------------------------------------------------------------------------- /internal/coord.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/coord.py -------------------------------------------------------------------------------- /internal/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/datasets.py -------------------------------------------------------------------------------- /internal/geopoly.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/geopoly.py -------------------------------------------------------------------------------- /internal/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/image.py -------------------------------------------------------------------------------- /internal/math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/math.py -------------------------------------------------------------------------------- /internal/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/models.py -------------------------------------------------------------------------------- /internal/pycolmap/.gitignore: -------------------------------------------------------------------------------- 1 | *.pyc 2 | *.sw* 3 | -------------------------------------------------------------------------------- /internal/pycolmap/LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/pycolmap/LICENSE.txt -------------------------------------------------------------------------------- /internal/pycolmap/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/pycolmap/README.md -------------------------------------------------------------------------------- /internal/pycolmap/pycolmap/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/pycolmap/pycolmap/__init__.py -------------------------------------------------------------------------------- /internal/pycolmap/pycolmap/camera.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/pycolmap/pycolmap/camera.py -------------------------------------------------------------------------------- /internal/pycolmap/pycolmap/database.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/pycolmap/pycolmap/database.py -------------------------------------------------------------------------------- /internal/pycolmap/pycolmap/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/pycolmap/pycolmap/image.py -------------------------------------------------------------------------------- /internal/pycolmap/pycolmap/rotation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/pycolmap/pycolmap/rotation.py -------------------------------------------------------------------------------- /internal/pycolmap/pycolmap/scene_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/pycolmap/pycolmap/scene_manager.py -------------------------------------------------------------------------------- /internal/pycolmap/tools/colmap_to_nvm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/pycolmap/tools/colmap_to_nvm.py -------------------------------------------------------------------------------- /internal/pycolmap/tools/delete_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/pycolmap/tools/delete_images.py -------------------------------------------------------------------------------- /internal/pycolmap/tools/impute_missing_cameras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/pycolmap/tools/impute_missing_cameras.py -------------------------------------------------------------------------------- /internal/pycolmap/tools/save_cameras_as_ply.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/pycolmap/tools/save_cameras_as_ply.py -------------------------------------------------------------------------------- /internal/pycolmap/tools/transform_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/pycolmap/tools/transform_model.py -------------------------------------------------------------------------------- /internal/pycolmap/tools/write_camera_track_to_bundler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/pycolmap/tools/write_camera_track_to_bundler.py -------------------------------------------------------------------------------- /internal/pycolmap/tools/write_depthmap_to_ply.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/pycolmap/tools/write_depthmap_to_ply.py -------------------------------------------------------------------------------- /internal/raw_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/raw_utils.py -------------------------------------------------------------------------------- /internal/ref_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/ref_utils.py -------------------------------------------------------------------------------- /internal/render.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/render.py -------------------------------------------------------------------------------- /internal/stepfun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/stepfun.py -------------------------------------------------------------------------------- /internal/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/train_utils.py -------------------------------------------------------------------------------- /internal/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/utils.py -------------------------------------------------------------------------------- /internal/vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/internal/vis.py -------------------------------------------------------------------------------- /render.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/render.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/render_neuropump.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/scripts/render_neuropump.sh -------------------------------------------------------------------------------- /scripts/render_neuropump_extension_change_background_light.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/scripts/render_neuropump_extension_change_background_light.sh -------------------------------------------------------------------------------- /scripts/render_neuropump_extension_change_refraction_index.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/scripts/render_neuropump_extension_change_refraction_index.sh -------------------------------------------------------------------------------- /scripts/render_neuropump_extension_change_s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/scripts/render_neuropump_extension_change_s.sh -------------------------------------------------------------------------------- /scripts/train_neuropump.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/scripts/train_neuropump.sh -------------------------------------------------------------------------------- /tests/camera_utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/tests/camera_utils_test.py -------------------------------------------------------------------------------- /tests/coord_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/tests/coord_test.py -------------------------------------------------------------------------------- /tests/datasets_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/tests/datasets_test.py -------------------------------------------------------------------------------- /tests/geopoly_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/tests/geopoly_test.py -------------------------------------------------------------------------------- /tests/image_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/tests/image_test.py -------------------------------------------------------------------------------- /tests/math_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/tests/math_test.py -------------------------------------------------------------------------------- /tests/ref_utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/tests/ref_utils_test.py -------------------------------------------------------------------------------- /tests/render_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/tests/render_test.py -------------------------------------------------------------------------------- /tests/stepfun_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/tests/stepfun_test.py -------------------------------------------------------------------------------- /tests/utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/tests/utils_test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YGswu/NeuroPump/HEAD/train.py --------------------------------------------------------------------------------