diff --git a/.github/workflows/code_checks.yml b/.github/workflows/code_checks.yml index 706576f75..4571ce8c5 100644 --- a/.github/workflows/code_checks.yml +++ b/.github/workflows/code_checks.yml @@ -25,10 +25,10 @@ jobs: pip install -r environment/requirements.txt pip install -e . - name: Run Black - run: black radiance/ notebooks/ --check + run: black pyrad/ notebooks/ --check - name: Python Pylint run: | - pylint radiance tests + pylint pyrad tests - name: Test with pytest run: | pytest diff --git a/README.md b/README.md index e12ef89b9..e2b9f4cbb 100644 --- a/README.md +++ b/README.md @@ -1,4 +1,4 @@ -# Radiance +# pyrad # Quickstart @@ -6,14 +6,14 @@ ``` # Clone the repo -git clone --recurse-submodules git@github.com:ethanweber/radiance.git +git clone --recurse-submodules git@github.com:ethanweber/pyrad.git # Create the python environment -conda create --name radiance python=3.8.13 -conda activate radiance +conda create --name pyrad python=3.8.13 +conda activate pyrad pip install -r environment/requirements.txt -# Install radiance as a library +# Install pyrad as a library pip install -e . # Install library with CUDA support. Change setup.py to `USE_CUDA = True` and then @@ -69,7 +69,7 @@ bash environments/run_jupyter.sh # Tooling -1. One can use [autoenv](https://github.com/hyperupcall/autoenv) to make setting up the environment and environment variables easier. This will run the `.envrc` file upon entering the `/path/to/radiance` folder. It will also remove the environment parameters upon leaving. +1. One can use [autoenv](https://github.com/hyperupcall/autoenv) to make setting up the environment and environment variables easier. This will run the `.envrc` file upon entering the `/path/to/pyrad` folder. It will also remove the environment parameters upon leaving. ``` # Install direnv. @@ -79,7 +79,7 @@ sudo apt install direnv eval "$(direnv hook bash)" # Populate your .envrc with commands you want to run. Then, run the following to allow updates. -cd /path/to/radiance +cd /path/to/pyrad direnv allow . ``` @@ -125,12 +125,12 @@ python scripts/benchmarking/run_benchmark.py ``` # nerf-pytorch cd external -python run_nerf.py --config configs/chair.txt --datadir /path/to/radiance/data/blender/chair +python run_nerf.py --config configs/chair.txt --datadir /path/to/pyrad/data/blender/chair # jaxnerf cd external conda activate jaxnerf -python -m jaxnerf.train --data_dir=/path/to/radiance/data/blender/chair --train_dir=/path/to/radiance/outputs/blender_chair_jaxnerf --config=/path/to/radiance/external/jaxnerf/configs/demo --render_every 100 +python -m jaxnerf.train --data_dir=/path/to/pyrad/data/blender/chair --train_dir=/path/to/pyrad/outputs/blender_chair_jaxnerf --config=/path/to/pyrad/external/jaxnerf/configs/demo --render_every 100 ``` # Speeding up the code diff --git a/configs/instant_ngp.yaml b/configs/instant_ngp.yaml index 01e3d5b80..6e3fe38ae 100644 --- a/configs/instant_ngp.yaml +++ b/configs/instant_ngp.yaml @@ -8,7 +8,7 @@ data.val_num_rays_per_chunk: 16384 graph: network: - _target_: radiance.nerf.graph.instant_ngp.NGPGraph + _target_: pyrad.nerf.graph.instant_ngp.NGPGraph field_implementation: torch # torch, tcnn, ... loss_coefficients: # loss coefficients rgb_loss: 1.0 diff --git a/configs/nerfw.yaml b/configs/nerfw.yaml index ce296fa9d..4d275b953 100644 --- a/configs/nerfw.yaml +++ b/configs/nerfw.yaml @@ -8,7 +8,7 @@ method_name: nerfw graph: network: - _target_: radiance.nerf.graph.nerfw.NerfWGraph + _target_: pyrad.nerf.graph.nerfw.NerfWGraph near_plane: 0.1 far_plane: 10.0 num_coarse_samples: 64 diff --git a/configs/semantic_nerf.yaml b/configs/semantic_nerf.yaml index a078cced6..c735fd04f 100644 --- a/configs/semantic_nerf.yaml +++ b/configs/semantic_nerf.yaml @@ -8,7 +8,7 @@ method_name: semantic_nerf graph: network: - _target_: radiance.nerf.graph.semantic_nerf.SemanticNerfGraph + _target_: pyrad.nerf.graph.semantic_nerf.SemanticNerfGraph near_plane: 0.1 far_plane: 10.0 num_coarse_samples: 64 diff --git a/configs/vanilla_nerf.yaml b/configs/vanilla_nerf.yaml index 56d63176e..b1d763619 100644 --- a/configs/vanilla_nerf.yaml +++ b/configs/vanilla_nerf.yaml @@ -7,7 +7,7 @@ method_name: vanilla_nerf graph: # saving model specifications - model_dir: /tmp/radiance_models/ + model_dir: /tmp/pyrad_models/ steps_per_save: null steps_per_test: 100 max_num_iterations: 1000000 @@ -18,7 +18,7 @@ graph: load_step: # pretrained model's step number network: - _target_: radiance.nerf.graph.vanilla_nerf.NeRFGraph + _target_: pyrad.nerf.graph.vanilla_nerf.NeRFGraph near_plane: 2.0 far_plane: 6.0 num_coarse_samples: 64 @@ -35,6 +35,6 @@ graph: _target_: torch.optim.RAdam lr: 0.0005 scheduler: - _target_: radiance.nerf.optimizers.ExponentialDecaySchedule + _target_: pyrad.nerf.optimizers.ExponentialDecaySchedule lr_final: 0.000005 max_steps: 1000000 diff --git a/docs/conf.py b/docs/conf.py index 1e8227c56..4aa57b313 100644 --- a/docs/conf.py +++ b/docs/conf.py @@ -17,12 +17,12 @@ # -- Project information ----------------------------------------------------- -project = 'Radiance' -copyright = '2022, Radiance Team' -author = 'Radiance Team' +project = "pyrad" +copyright = "2022, pyrad Team" +author = "pyrad Team" # The full version, including alpha/beta/rc tags -release = '0.0.0' +release = "0.0.0" # -- General configuration --------------------------------------------------- @@ -30,16 +30,15 @@ release = '0.0.0' # Add any Sphinx extension module names here, as strings. They can be # extensions coming with Sphinx (named 'sphinx.ext.*') or your custom # ones. -extensions = [ -] +extensions = [] # Add any paths that contain templates here, relative to this directory. -templates_path = ['_templates'] +templates_path = ["_templates"] # List of patterns, relative to source directory, that match files and # directories to ignore when looking for source files. # This pattern also affects html_static_path and html_extra_path. -exclude_patterns = ['_build', 'Thumbs.db', '.DS_Store'] +exclude_patterns = ["_build", "Thumbs.db", ".DS_Store"] # -- Options for HTML output ------------------------------------------------- @@ -47,9 +46,9 @@ exclude_patterns = ['_build', 'Thumbs.db', '.DS_Store'] # The theme to use for HTML and HTML Help pages. See the documentation for # a list of builtin themes. # -html_theme = 'alabaster' +html_theme = "alabaster" # Add any paths that contain custom static files (such as style sheets) here, # relative to this directory. They are copied after the builtin static files, # so a file named "default.css" will overwrite the builtin "default.css". -html_static_path = ['_static'] \ No newline at end of file +html_static_path = ["_static"] diff --git a/docs/index.rst b/docs/index.rst index 72731d572..0902c6ce8 100644 --- a/docs/index.rst +++ b/docs/index.rst @@ -1,9 +1,9 @@ -.. Radiance documentation master file, created by +.. pyrad documentation master file, created by sphinx-quickstart on Tue May 31 11:49:30 2022. You can adapt this file completely to your liking, but it should at least contain the root `toctree` directive. -Welcome to Radiance's documentation! +Welcome to pyrad's documentation! ==================================== .. toctree:: diff --git a/notebooks/visualize_cameras.ipynb b/notebooks/visualize_cameras.ipynb index 2a0320f51..07f09c1e8 100644 --- a/notebooks/visualize_cameras.ipynb +++ b/notebooks/visualize_cameras.ipynb @@ -14,7 +14,7 @@ "outputs": [], "source": [ "import torch\n", - "from radiance.structures import cameras\n", + "from pyrad.structures import cameras\n", "import plotly.graph_objects as go\n", "import mediapy as media" ] @@ -19597,7 +19597,7 @@ "hash": "024fe9dea615e4e34961116ff517c7e9f5910440e3d2c8eb4560299f11aaf2e7" }, "kernelspec": { - "display_name": "Python 3.9.12 ('radiance')", + "display_name": "Python 3.9.12 ('pyrad')", "language": "python", "name": "python3" }, diff --git a/notebooks/visualize_dataset.ipynb b/notebooks/visualize_dataset.ipynb index d664ed738..b89563d65 100644 --- a/notebooks/visualize_dataset.ipynb +++ b/notebooks/visualize_dataset.ipynb @@ -20,15 +20,15 @@ "import torch\n", "from torch.utils.data import DataLoader\n", "\n", - "from radiance.nerf.dataset.image_dataset import ImageDataset, collate_batch\n", - "from radiance.nerf.dataset.collate import CollateIterDataset, collate_batch_size_one\n", - "from radiance.nerf.dataset.utils import get_dataset_inputs\n", - "from radiance.nerf.field_modules.ray_generator import RayGenerator\n", - "from radiance.nerf.field_modules.scene_bounds_collider import SceneBoundsCollider\n", - "from radiance.structures.rays import RayBundle\n", - "from radiance.utils.io import get_absolute_path\n", - "from radiance.viewer.plotly import get_line_segments_from_lines\n", - "from radiance.structures.cameras import get_camera_model\n", + "from pyrad.nerf.dataset.image_dataset import ImageDataset, collate_batch\n", + "from pyrad.nerf.dataset.collate import CollateIterDataset, collate_batch_size_one\n", + "from pyrad.nerf.dataset.utils import get_dataset_inputs\n", + "from pyrad.nerf.field_modules.ray_generator import RayGenerator\n", + "from pyrad.nerf.field_modules.scene_bounds_collider import SceneBoundsCollider\n", + "from pyrad.structures.rays import RayBundle\n", + "from pyrad.utils.io import get_absolute_path\n", + "from pyrad.viewer.plotly import get_line_segments_from_lines\n", + "from pyrad.structures.cameras import get_camera_model\n", "\n", "\n", "from hydra import compose, initialize\n", @@ -361,9 +361,9 @@ "hash": "4a749acc7d255f078aee52e0584cc77b3cb5aaed1b3a7407ec4262c1bf6cb526" }, "kernelspec": { - "display_name": "radiance", + "display_name": "pyrad", "language": "python", - "name": "radiance" + "name": "pyrad" }, "language_info": { "codemirror_mode": { diff --git a/notebooks/visualize_encoders.ipynb b/notebooks/visualize_encoders.ipynb index 118a41eb2..b09536150 100644 --- a/notebooks/visualize_encoders.ipynb +++ b/notebooks/visualize_encoders.ipynb @@ -14,7 +14,7 @@ "outputs": [ { "data": { - "application/javascript": "\n setTimeout(function() {\n var nbb_cell_id = 6;\n var nbb_unformatted_code = \"import torch\\nimport mediapy as media\\nfrom radiance.nerf.field_modules import encoding\";\n var nbb_formatted_code = \"import torch\\nimport mediapy as media\\nfrom radiance.nerf.field_modules import encoding\";\n var nbb_cells = Jupyter.notebook.get_cells();\n for (var i = 0; i < nbb_cells.length; ++i) {\n if (nbb_cells[i].input_prompt_number == nbb_cell_id) {\n if (nbb_cells[i].get_text() == nbb_unformatted_code) {\n nbb_cells[i].set_text(nbb_formatted_code);\n }\n break;\n }\n }\n }, 500);\n ", + "application/javascript": "\n setTimeout(function() {\n var nbb_cell_id = 6;\n var nbb_unformatted_code = \"import torch\\nimport mediapy as media\\nfrom pyrad.nerf.field_modules import encoding\";\n var nbb_formatted_code = \"import torch\\nimport mediapy as media\\nfrom pyrad.nerf.field_modules import encoding\";\n var nbb_cells = Jupyter.notebook.get_cells();\n for (var i = 0; i < nbb_cells.length; ++i) {\n if (nbb_cells[i].input_prompt_number == nbb_cell_id) {\n if (nbb_cells[i].get_text() == nbb_unformatted_code) {\n nbb_cells[i].set_text(nbb_formatted_code);\n }\n break;\n }\n }\n }, 500);\n ", "text/plain": [ "" ] @@ -26,7 +26,7 @@ "source": [ "import torch\n", "import mediapy as media\n", - "from radiance.nerf.field_modules import encoding" + "from pyrad.nerf.field_modules import encoding" ] }, { @@ -434,7 +434,7 @@ "hash": "024fe9dea615e4e34961116ff517c7e9f5910440e3d2c8eb4560299f11aaf2e7" }, "kernelspec": { - "display_name": "Python 3.9.11 ('radiance')", + "display_name": "Python 3.9.11 ('pyrad')", "language": "python", "name": "python3" }, diff --git a/pyproject.toml b/pyproject.toml index 8afbd775c..33bc605bd 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -11,7 +11,7 @@ line-length = 120 max-line-length = 120 generated-members = ["numpy.*", "torch.*", "cv2.*", "cv.*"] good-names-rgxs = "^[_a-zA-Z][_a-z0-9]?$" -ignore = ["radiance/cuda", "tests/cuda", "nerfw.py", "semantic_nerf.py"] +ignore = ["pyrad/cuda", "tests/cuda", "nerfw.py", "semantic_nerf.py"] disable = [ "too-many-arguments", @@ -23,7 +23,7 @@ disable = [ #pytest [tool.pytest.ini_options] -addopts = "--typeguard-packages=radiance --torchtyping-patch-typeguard --disable-warnings --ignore=tests/cuda" +addopts = "--typeguard-packages=pyrad --torchtyping-patch-typeguard --disable-warnings --ignore=tests/cuda" testpaths = [ "tests", ] \ No newline at end of file diff --git a/radiance/cuda/include/helpers.cuh b/pyrad/cuda/include/helpers.cuh similarity index 100% rename from radiance/cuda/include/helpers.cuh rename to pyrad/cuda/include/helpers.cuh diff --git a/radiance/cuda/mattport_cuda.cpp b/pyrad/cuda/mattport_cuda.cpp similarity index 100% rename from radiance/cuda/mattport_cuda.cpp rename to pyrad/cuda/mattport_cuda.cpp diff --git a/radiance/cuda/mattport_cuda_kernel.cu b/pyrad/cuda/mattport_cuda_kernel.cu similarity index 100% rename from radiance/cuda/mattport_cuda_kernel.cu rename to pyrad/cuda/mattport_cuda_kernel.cu diff --git a/radiance/cuda/sampler.py b/pyrad/cuda/sampler.py similarity index 85% rename from radiance/cuda/sampler.py rename to pyrad/cuda/sampler.py index 635a7d981..c7a657056 100644 --- a/radiance/cuda/sampler.py +++ b/pyrad/cuda/sampler.py @@ -7,13 +7,13 @@ import math import torch from torch import nn -import radiance_cuda +import pyrad_cuda class LLTMFunction(torch.autograd.Function): @staticmethod def forward(ctx, input, weights, bias, old_h, old_cell): - outputs = radiance_cuda.forward(input, weights, bias, old_h, old_cell) + outputs = pyrad_cuda.forward(input, weights, bias, old_h, old_cell) new_h, new_cell = outputs[:2] variables = outputs[1:] + [weights] ctx.save_for_backward(*variables) @@ -22,7 +22,7 @@ class LLTMFunction(torch.autograd.Function): @staticmethod def backward(ctx, grad_h, grad_cell): - outputs = radiance_cuda.backward(grad_h.contiguous(), grad_cell.contiguous(), *ctx.saved_variables) + outputs = pyrad_cuda.backward(grad_h.contiguous(), grad_cell.contiguous(), *ctx.saved_variables) d_old_h, d_input, d_weights, d_bias, d_old_cell, d_gates = outputs return d_input, d_weights, d_bias, d_old_h, d_old_cell diff --git a/radiance/nerf/__init__.py b/pyrad/nerf/__init__.py similarity index 100% rename from radiance/nerf/__init__.py rename to pyrad/nerf/__init__.py diff --git a/radiance/nerf/colliders.py b/pyrad/nerf/colliders.py similarity index 96% rename from radiance/nerf/colliders.py rename to pyrad/nerf/colliders.py index 57a192774..15cb3e737 100644 --- a/radiance/nerf/colliders.py +++ b/pyrad/nerf/colliders.py @@ -4,8 +4,8 @@ Ray generator. import torch from torch import nn from torchtyping import TensorType -from radiance.nerf.dataset.structs import SceneBounds -from radiance.structures.rays import RayBundle +from pyrad.nerf.dataset.structs import SceneBounds +from pyrad.structures.rays import RayBundle class SceneBoundsCollider(nn.Module): diff --git a/radiance/nerf/dataset/__init__.py b/pyrad/nerf/dataset/__init__.py similarity index 100% rename from radiance/nerf/dataset/__init__.py rename to pyrad/nerf/dataset/__init__.py diff --git a/radiance/nerf/dataset/collate.py b/pyrad/nerf/dataset/collate.py similarity index 100% rename from radiance/nerf/dataset/collate.py rename to pyrad/nerf/dataset/collate.py diff --git a/radiance/nerf/dataset/colmap_utils.py b/pyrad/nerf/dataset/colmap_utils.py similarity index 100% rename from radiance/nerf/dataset/colmap_utils.py rename to pyrad/nerf/dataset/colmap_utils.py diff --git a/radiance/nerf/dataset/format/__init__.py b/pyrad/nerf/dataset/format/__init__.py similarity index 100% rename from radiance/nerf/dataset/format/__init__.py rename to pyrad/nerf/dataset/format/__init__.py diff --git a/radiance/nerf/dataset/format/blender.py b/pyrad/nerf/dataset/format/blender.py similarity index 95% rename from radiance/nerf/dataset/format/blender.py rename to pyrad/nerf/dataset/format/blender.py index deea8901e..f436f8bc2 100644 --- a/radiance/nerf/dataset/format/blender.py +++ b/pyrad/nerf/dataset/format/blender.py @@ -10,8 +10,8 @@ import numpy as np import torch from torchtyping import TensorType -from radiance.nerf.dataset.structs import DatasetInputs, SceneBounds -from radiance.utils.io import load_from_json +from pyrad.nerf.dataset.structs import DatasetInputs, SceneBounds +from pyrad.utils.io import load_from_json def load_blender_data( diff --git a/radiance/nerf/dataset/format/friends.py b/pyrad/nerf/dataset/format/friends.py similarity index 95% rename from radiance/nerf/dataset/format/friends.py rename to pyrad/nerf/dataset/format/friends.py index 26ca641d1..4d3140832 100644 --- a/radiance/nerf/dataset/format/friends.py +++ b/pyrad/nerf/dataset/format/friends.py @@ -7,10 +7,10 @@ import numpy as np import torch -from radiance.nerf.dataset.colmap_utils import read_cameras_binary, read_images_binary, read_pointsTD_binary -from radiance.nerf.dataset.structs import DatasetInputs, PointCloud, SceneBounds, Semantics -from radiance.utils import profiler -from radiance.utils.io import load_from_json +from pyrad.nerf.dataset.colmap_utils import read_cameras_binary, read_images_binary, read_pointsTD_binary +from pyrad.nerf.dataset.structs import DatasetInputs, PointCloud, SceneBounds, Semantics +from pyrad.utils import profiler +from pyrad.utils.io import load_from_json def get_aabb_and_transform(basedir): diff --git a/radiance/nerf/dataset/format/instant_ngp.py b/pyrad/nerf/dataset/format/instant_ngp.py similarity index 95% rename from radiance/nerf/dataset/format/instant_ngp.py rename to pyrad/nerf/dataset/format/instant_ngp.py index 43f3378ef..d645d9a94 100644 --- a/radiance/nerf/dataset/format/instant_ngp.py +++ b/pyrad/nerf/dataset/format/instant_ngp.py @@ -10,8 +10,8 @@ import imageio import numpy as np import torch -from radiance.nerf.dataset.structs import DatasetInputs, SceneBounds -from radiance.utils.io import load_from_json +from pyrad.nerf.dataset.structs import DatasetInputs, SceneBounds +from pyrad.utils.io import load_from_json def load_instant_ngp_data( diff --git a/radiance/nerf/dataset/format/llff.py b/pyrad/nerf/dataset/format/llff.py similarity index 100% rename from radiance/nerf/dataset/format/llff.py rename to pyrad/nerf/dataset/format/llff.py diff --git a/radiance/nerf/dataset/image_dataset.py b/pyrad/nerf/dataset/image_dataset.py similarity index 99% rename from radiance/nerf/dataset/image_dataset.py rename to pyrad/nerf/dataset/image_dataset.py index f37d9ead3..9a28330cf 100644 --- a/radiance/nerf/dataset/image_dataset.py +++ b/pyrad/nerf/dataset/image_dataset.py @@ -12,7 +12,7 @@ from PIL import Image from torch.utils.data import default_collate from torchtyping import TensorType -from radiance.nerf.dataset.structs import Semantics +from pyrad.nerf.dataset.structs import Semantics def collate_batch(batch_list, num_rays_per_batch, keep_full_image: bool = False): diff --git a/radiance/nerf/dataset/structs.py b/pyrad/nerf/dataset/structs.py similarity index 100% rename from radiance/nerf/dataset/structs.py rename to pyrad/nerf/dataset/structs.py diff --git a/radiance/nerf/dataset/utils.py b/pyrad/nerf/dataset/utils.py similarity index 85% rename from radiance/nerf/dataset/utils.py rename to pyrad/nerf/dataset/utils.py index 8fead05e6..aa8ad92cb 100644 --- a/radiance/nerf/dataset/utils.py +++ b/pyrad/nerf/dataset/utils.py @@ -6,12 +6,12 @@ from typing import Optional, Union from omegaconf import ListConfig -from radiance.nerf.dataset.format.blender import load_blender_data -from radiance.nerf.dataset.format.friends import load_friends_data -from radiance.nerf.dataset.format.instant_ngp import load_instant_ngp_data -from radiance.nerf.dataset.structs import DatasetInputs -from radiance.structures.colors import get_color -from radiance.utils.io import get_absolute_path +from pyrad.nerf.dataset.format.blender import load_blender_data +from pyrad.nerf.dataset.format.friends import load_friends_data +from pyrad.nerf.dataset.format.instant_ngp import load_instant_ngp_data +from pyrad.nerf.dataset.structs import DatasetInputs +from pyrad.structures.colors import get_color +from pyrad.utils.io import get_absolute_path def get_dataset_inputs( diff --git a/radiance/nerf/field_modules/__init__.py b/pyrad/nerf/field_modules/__init__.py similarity index 100% rename from radiance/nerf/field_modules/__init__.py rename to pyrad/nerf/field_modules/__init__.py diff --git a/radiance/nerf/field_modules/base.py b/pyrad/nerf/field_modules/base.py similarity index 100% rename from radiance/nerf/field_modules/base.py rename to pyrad/nerf/field_modules/base.py diff --git a/radiance/nerf/field_modules/embedding.py b/pyrad/nerf/field_modules/embedding.py similarity index 93% rename from radiance/nerf/field_modules/embedding.py rename to pyrad/nerf/field_modules/embedding.py index f6140f40c..d265a0dc5 100644 --- a/radiance/nerf/field_modules/embedding.py +++ b/pyrad/nerf/field_modules/embedding.py @@ -5,7 +5,7 @@ Code for embeddings. import torch from torchtyping import TensorType -from radiance.nerf.field_modules.base import FieldModule +from pyrad.nerf.field_modules.base import FieldModule class Embedding(FieldModule): diff --git a/radiance/nerf/field_modules/encoding.py b/pyrad/nerf/field_modules/encoding.py similarity index 99% rename from radiance/nerf/field_modules/encoding.py rename to pyrad/nerf/field_modules/encoding.py index 91e88fe79..1632030ea 100644 --- a/radiance/nerf/field_modules/encoding.py +++ b/pyrad/nerf/field_modules/encoding.py @@ -9,8 +9,8 @@ import torch.nn.functional as F from torch import nn from torchtyping import TensorType -from radiance.nerf.field_modules.base import FieldModule -from radiance.utils.math import components_from_spherical_harmonics +from pyrad.nerf.field_modules.base import FieldModule +from pyrad.utils.math import components_from_spherical_harmonics class Encoding(FieldModule): diff --git a/radiance/nerf/field_modules/field_heads.py b/pyrad/nerf/field_modules/field_heads.py similarity index 98% rename from radiance/nerf/field_modules/field_heads.py rename to pyrad/nerf/field_modules/field_heads.py index e974e8f5e..99b18bff7 100644 --- a/radiance/nerf/field_modules/field_heads.py +++ b/pyrad/nerf/field_modules/field_heads.py @@ -7,7 +7,7 @@ from typing import Optional from torch import nn from torchtyping import TensorType -from radiance.nerf.field_modules.base import FieldModule +from pyrad.nerf.field_modules.base import FieldModule class FieldHeadNames(Enum): diff --git a/radiance/nerf/field_modules/field_losses.py b/pyrad/nerf/field_modules/field_losses.py similarity index 100% rename from radiance/nerf/field_modules/field_losses.py rename to pyrad/nerf/field_modules/field_losses.py diff --git a/radiance/nerf/field_modules/mlp.py b/pyrad/nerf/field_modules/mlp.py similarity index 98% rename from radiance/nerf/field_modules/mlp.py rename to pyrad/nerf/field_modules/mlp.py index 16c9f9350..b70fc5eb2 100644 --- a/radiance/nerf/field_modules/mlp.py +++ b/pyrad/nerf/field_modules/mlp.py @@ -7,7 +7,7 @@ import torch from torch import nn from torchtyping import TensorType -from radiance.nerf.field_modules.base import FieldModule +from pyrad.nerf.field_modules.base import FieldModule class MLP(FieldModule): diff --git a/radiance/nerf/fields/__init__.py b/pyrad/nerf/fields/__init__.py similarity index 100% rename from radiance/nerf/fields/__init__.py rename to pyrad/nerf/fields/__init__.py diff --git a/radiance/nerf/fields/base.py b/pyrad/nerf/fields/base.py similarity index 94% rename from radiance/nerf/fields/base.py rename to pyrad/nerf/fields/base.py index 1aa1a0221..8b7a986e0 100644 --- a/radiance/nerf/fields/base.py +++ b/pyrad/nerf/fields/base.py @@ -8,10 +8,10 @@ from typing import Dict, Optional, Tuple from torch import nn import torch from torchtyping import TensorType -from radiance.nerf.field_modules.field_heads import FieldHeadNames +from pyrad.nerf.field_modules.field_heads import FieldHeadNames -from radiance.structures.rays import PointSamples -from radiance.utils.misc import is_not_none +from pyrad.structures.rays import PointSamples +from pyrad.utils.misc import is_not_none class Field(nn.Module): diff --git a/radiance/nerf/fields/instant_ngp_field.py b/pyrad/nerf/fields/instant_ngp_field.py similarity index 93% rename from radiance/nerf/fields/instant_ngp_field.py rename to pyrad/nerf/fields/instant_ngp_field.py index 85d162780..ec9a184bc 100644 --- a/radiance/nerf/fields/instant_ngp_field.py +++ b/pyrad/nerf/fields/instant_ngp_field.py @@ -9,12 +9,12 @@ import torch import torch.nn.functional as F from torch.nn.parameter import Parameter -from radiance.nerf.field_modules.encoding import Encoding, HashEncoding, SHEncoding -from radiance.nerf.field_modules.field_heads import FieldHeadNames -from radiance.nerf.fields.base import Field -from radiance.nerf.fields.nerf_field import NeRFField -from radiance.structures.rays import PointSamples -from radiance.utils.misc import is_not_none +from pyrad.nerf.field_modules.encoding import Encoding, HashEncoding, SHEncoding +from pyrad.nerf.field_modules.field_heads import FieldHeadNames +from pyrad.nerf.fields.base import Field +from pyrad.nerf.fields.nerf_field import NeRFField +from pyrad.structures.rays import PointSamples +from pyrad.utils.misc import is_not_none try: import tinycudann as tcnn diff --git a/radiance/nerf/fields/nerf_field.py b/pyrad/nerf/fields/nerf_field.py similarity index 90% rename from radiance/nerf/fields/nerf_field.py rename to pyrad/nerf/fields/nerf_field.py index 6325025fe..9a4b7589a 100644 --- a/radiance/nerf/fields/nerf_field.py +++ b/pyrad/nerf/fields/nerf_field.py @@ -7,11 +7,11 @@ import torch from torch import nn from torchtyping import TensorType -from radiance.nerf.field_modules.encoding import Encoding, Identity -from radiance.nerf.field_modules.field_heads import DensityFieldHead, FieldHead, FieldHeadNames, RGBFieldHead -from radiance.nerf.field_modules.mlp import MLP -from radiance.nerf.fields.base import Field -from radiance.structures.rays import PointSamples +from pyrad.nerf.field_modules.encoding import Encoding, Identity +from pyrad.nerf.field_modules.field_heads import DensityFieldHead, FieldHead, FieldHeadNames, RGBFieldHead +from pyrad.nerf.field_modules.mlp import MLP +from pyrad.nerf.fields.base import Field +from pyrad.structures.rays import PointSamples class NeRFField(Field): diff --git a/radiance/nerf/graph/base.py b/pyrad/nerf/graph/base.py similarity index 94% rename from radiance/nerf/graph/base.py rename to pyrad/nerf/graph/base.py index c91735029..8e05bdd8e 100644 --- a/radiance/nerf/graph/base.py +++ b/pyrad/nerf/graph/base.py @@ -11,13 +11,13 @@ from torch import nn from torch.nn import Parameter from torchtyping import TensorType -from radiance.nerf.dataset.structs import SceneBounds -from radiance.nerf.ray_generator import RayGenerator -from radiance.nerf.colliders import AABBBoxCollider -from radiance.structures.cameras import get_camera_model -from radiance.structures.rays import RayBundle -from radiance.utils.misc import get_masked_dict -from radiance.utils.misc import is_not_none +from pyrad.nerf.dataset.structs import SceneBounds +from pyrad.nerf.ray_generator import RayGenerator +from pyrad.nerf.colliders import AABBBoxCollider +from pyrad.structures.cameras import get_camera_model +from pyrad.structures.rays import RayBundle +from pyrad.utils.misc import get_masked_dict +from pyrad.utils.misc import is_not_none class AbstractGraph(nn.Module): diff --git a/radiance/nerf/graph/instant_ngp.py b/pyrad/nerf/graph/instant_ngp.py similarity index 90% rename from radiance/nerf/graph/instant_ngp.py rename to pyrad/nerf/graph/instant_ngp.py index 3c033a515..5e2a4ee65 100644 --- a/radiance/nerf/graph/instant_ngp.py +++ b/pyrad/nerf/graph/instant_ngp.py @@ -9,17 +9,17 @@ from torch.nn import Parameter from torchmetrics import PeakSignalNoiseRatio, StructuralSimilarityIndexMeasure from torchmetrics.image.lpip import LearnedPerceptualImagePatchSimilarity -from radiance.nerf.colliders import AABBBoxCollider -from radiance.nerf.field_modules.field_heads import FieldHeadNames -from radiance.nerf.fields.instant_ngp_field import field_implementation_to_class -from radiance.nerf.graph.base import Graph -from radiance.nerf.loss import MSELoss -from radiance.nerf.occupancy_grid import OccupancyGrid -from radiance.nerf.renderers import AccumulationRenderer, DepthRenderer, RGBRenderer -from radiance.nerf.sampler import PDFSampler, UniformSampler -from radiance.structures import colors -from radiance.structures.rays import RayBundle -from radiance.utils import visualization, writer +from pyrad.nerf.colliders import AABBBoxCollider +from pyrad.nerf.field_modules.field_heads import FieldHeadNames +from pyrad.nerf.fields.instant_ngp_field import field_implementation_to_class +from pyrad.nerf.graph.base import Graph +from pyrad.nerf.loss import MSELoss +from pyrad.nerf.occupancy_grid import OccupancyGrid +from pyrad.nerf.renderers import AccumulationRenderer, DepthRenderer, RGBRenderer +from pyrad.nerf.sampler import PDFSampler, UniformSampler +from pyrad.structures import colors +from pyrad.structures.rays import RayBundle +from pyrad.utils import visualization, writer class NGPGraph(Graph): diff --git a/radiance/nerf/graph/nerfw.py b/pyrad/nerf/graph/nerfw.py similarity index 95% rename from radiance/nerf/graph/nerfw.py rename to pyrad/nerf/graph/nerfw.py index 42cf93335..43acd7554 100644 --- a/radiance/nerf/graph/nerfw.py +++ b/pyrad/nerf/graph/nerfw.py @@ -8,18 +8,18 @@ from typing import Tuple import torch from torch import nn -from radiance.nerf.field_modules.embedding import Embedding -from radiance.nerf.field_modules.field_heads import ( +from pyrad.nerf.field_modules.embedding import Embedding +from pyrad.nerf.field_modules.field_heads import ( FieldHeadNames, TransientDensityHead, TransientRGBHead, UncertaintyFieldHead, ) -from radiance.nerf.field_modules.mlp import MLP -from radiance.nerf.graph.vanilla_nerf import NeRFField, NeRFGraph -from radiance.nerf.renderers import UncertaintyRenderer -from radiance.structures.rays import RayBundle, RaySamples -from radiance.utils import visualization, writer +from pyrad.nerf.field_modules.mlp import MLP +from pyrad.nerf.graph.vanilla_nerf import NeRFField, NeRFGraph +from pyrad.nerf.renderers import UncertaintyRenderer +from pyrad.structures.rays import RayBundle, RaySamples +from pyrad.utils import visualization, writer class NerfWField(Field): diff --git a/radiance/nerf/graph/semantic_nerf.py b/pyrad/nerf/graph/semantic_nerf.py similarity index 94% rename from radiance/nerf/graph/semantic_nerf.py rename to pyrad/nerf/graph/semantic_nerf.py index 9ffb93136..ead8e0b4a 100644 --- a/radiance/nerf/graph/semantic_nerf.py +++ b/pyrad/nerf/graph/semantic_nerf.py @@ -7,12 +7,12 @@ from typing import Tuple import torch from torch import nn -from radiance.nerf.field_modules.field_heads import FieldHeadNames, SemanticStuffHead -from radiance.nerf.field_modules.mlp import MLP -from radiance.nerf.graph.vanilla_nerf import NeRFField, NeRFGraph -from radiance.nerf.renderers import SemanticRenderer -from radiance.structures.rays import RayBundle, RaySamples -from radiance.utils import visualization, writer +from pyrad.nerf.field_modules.field_heads import FieldHeadNames, SemanticStuffHead +from pyrad.nerf.field_modules.mlp import MLP +from pyrad.nerf.graph.vanilla_nerf import NeRFField, NeRFGraph +from pyrad.nerf.renderers import SemanticRenderer +from pyrad.structures.rays import RayBundle, RaySamples +from pyrad.utils import visualization, writer class SemanticNerfField(NeRFField): diff --git a/radiance/nerf/graph/tensorf.py b/pyrad/nerf/graph/tensorf.py similarity index 100% rename from radiance/nerf/graph/tensorf.py rename to pyrad/nerf/graph/tensorf.py diff --git a/radiance/nerf/graph/vanilla_nerf.py b/pyrad/nerf/graph/vanilla_nerf.py similarity index 92% rename from radiance/nerf/graph/vanilla_nerf.py rename to pyrad/nerf/graph/vanilla_nerf.py index 5e00795d9..c8421f4b9 100644 --- a/radiance/nerf/graph/vanilla_nerf.py +++ b/pyrad/nerf/graph/vanilla_nerf.py @@ -9,18 +9,18 @@ import torch from torch.nn import Parameter from torchmetrics import PeakSignalNoiseRatio, StructuralSimilarityIndexMeasure from torchmetrics.image.lpip import LearnedPerceptualImagePatchSimilarity -from radiance.nerf.field_modules.encoding import NeRFEncoding -from radiance.nerf.field_modules.field_heads import FieldHeadNames +from pyrad.nerf.field_modules.encoding import NeRFEncoding +from pyrad.nerf.field_modules.field_heads import FieldHeadNames -from radiance.nerf.fields.nerf_field import NeRFField -from radiance.nerf.graph.base import Graph -from radiance.nerf.loss import MSELoss -from radiance.nerf.renderers import AccumulationRenderer, DepthRenderer, RGBRenderer -from radiance.nerf.sampler import PDFSampler, UniformSampler -from radiance.structures import colors -from radiance.structures.rays import RayBundle -from radiance.nerf.colliders import NearFarCollider -from radiance.utils import visualization, writer +from pyrad.nerf.fields.nerf_field import NeRFField +from pyrad.nerf.graph.base import Graph +from pyrad.nerf.loss import MSELoss +from pyrad.nerf.renderers import AccumulationRenderer, DepthRenderer, RGBRenderer +from pyrad.nerf.sampler import PDFSampler, UniformSampler +from pyrad.structures import colors +from pyrad.structures.rays import RayBundle +from pyrad.nerf.colliders import NearFarCollider +from pyrad.utils import visualization, writer class NeRFGraph(Graph): diff --git a/radiance/nerf/loss.py b/pyrad/nerf/loss.py similarity index 100% rename from radiance/nerf/loss.py rename to pyrad/nerf/loss.py diff --git a/radiance/nerf/occupancy_grid.py b/pyrad/nerf/occupancy_grid.py similarity index 99% rename from radiance/nerf/occupancy_grid.py rename to pyrad/nerf/occupancy_grid.py index 87ed5920a..517805fc3 100644 --- a/radiance/nerf/occupancy_grid.py +++ b/pyrad/nerf/occupancy_grid.py @@ -7,7 +7,7 @@ import torch.nn.functional as F from torch import nn from torchtyping import TensorType -from radiance.utils.misc import is_not_none +from pyrad.utils.misc import is_not_none class OccupancyGrid(nn.Module): diff --git a/radiance/nerf/optimizers.py b/pyrad/nerf/optimizers.py similarity index 99% rename from radiance/nerf/optimizers.py rename to pyrad/nerf/optimizers.py index cddd02288..e0b0497df 100644 --- a/radiance/nerf/optimizers.py +++ b/pyrad/nerf/optimizers.py @@ -10,7 +10,7 @@ from omegaconf import DictConfig from torch.nn import Parameter from torch.optim.lr_scheduler import LambdaLR -from radiance.utils import writer +from pyrad.utils import writer class ExponentialDecaySchedule(LambdaLR): diff --git a/radiance/nerf/ray_generator.py b/pyrad/nerf/ray_generator.py similarity index 95% rename from radiance/nerf/ray_generator.py rename to pyrad/nerf/ray_generator.py index eb10f6f7c..23516695a 100644 --- a/radiance/nerf/ray_generator.py +++ b/pyrad/nerf/ray_generator.py @@ -3,8 +3,8 @@ Ray generator. """ from torch import nn from torchtyping import TensorType -from radiance.structures.rays import RayBundle -from radiance.structures.cameras import get_camera_model +from pyrad.structures.rays import RayBundle +from pyrad.structures.cameras import get_camera_model class RayGenerator(nn.Module): diff --git a/radiance/nerf/renderers.py b/pyrad/nerf/renderers.py similarity index 98% rename from radiance/nerf/renderers.py rename to pyrad/nerf/renderers.py index 91b630b59..05398a0d9 100644 --- a/radiance/nerf/renderers.py +++ b/pyrad/nerf/renderers.py @@ -8,7 +8,7 @@ import torch from torch import nn from torchtyping import TensorType -from radiance.utils.math import components_from_spherical_harmonics +from pyrad.utils.math import components_from_spherical_harmonics class RGBRenderer(nn.Module): diff --git a/radiance/nerf/sampler.py b/pyrad/nerf/sampler.py similarity index 98% rename from radiance/nerf/sampler.py rename to pyrad/nerf/sampler.py index 4f77f305f..33bf7a581 100644 --- a/radiance/nerf/sampler.py +++ b/pyrad/nerf/sampler.py @@ -9,8 +9,8 @@ import torch from torch import nn from torchtyping import TensorType -from radiance.nerf.occupancy_grid import OccupancyGrid -from radiance.structures.rays import RayBundle, RaySamples +from pyrad.nerf.occupancy_grid import OccupancyGrid +from pyrad.structures.rays import RayBundle, RaySamples class Sampler(nn.Module): diff --git a/radiance/nerf/trainer.py b/pyrad/nerf/trainer.py similarity index 96% rename from radiance/nerf/trainer.py rename to pyrad/nerf/trainer.py index 2457f2458..1c9bc1ebc 100644 --- a/radiance/nerf/trainer.py +++ b/pyrad/nerf/trainer.py @@ -14,14 +14,14 @@ from omegaconf import DictConfig from torch.nn.parallel import DistributedDataParallel as DDP from torch.utils.data import DataLoader -from radiance.nerf.dataset.collate import CollateIterDataset, collate_batch_size_one -from radiance.nerf.dataset.image_dataset import ImageDataset, collate_batch -from radiance.nerf.dataset.utils import DatasetInputs, get_dataset_inputs -from radiance.nerf.optimizers import Optimizers -from radiance.utils import profiler, writer -from radiance.utils.callbacks import update_occupancy -from radiance.utils.decorators import check_main_thread -from radiance.utils.misc import get_dict_to_torch +from pyrad.nerf.dataset.collate import CollateIterDataset, collate_batch_size_one +from pyrad.nerf.dataset.image_dataset import ImageDataset, collate_batch +from pyrad.nerf.dataset.utils import DatasetInputs, get_dataset_inputs +from pyrad.nerf.optimizers import Optimizers +from pyrad.utils import profiler, writer +from pyrad.utils.callbacks import update_occupancy +from pyrad.utils.decorators import check_main_thread +from pyrad.utils.misc import get_dict_to_torch logging.getLogger("PIL").setLevel(logging.WARNING) diff --git a/radiance/sfm/__init__.py b/pyrad/sfm/__init__.py similarity index 100% rename from radiance/sfm/__init__.py rename to pyrad/sfm/__init__.py diff --git a/radiance/structures/__init__.py b/pyrad/structures/__init__.py similarity index 100% rename from radiance/structures/__init__.py rename to pyrad/structures/__init__.py diff --git a/radiance/structures/cameras.py b/pyrad/structures/cameras.py similarity index 99% rename from radiance/structures/cameras.py rename to pyrad/structures/cameras.py index 5c2cfbd30..24ad6ebde 100644 --- a/radiance/structures/cameras.py +++ b/pyrad/structures/cameras.py @@ -9,7 +9,7 @@ import torch from torch.nn.functional import normalize from torchtyping import TensorType -from radiance.structures.rays import RayBundle +from pyrad.structures.rays import RayBundle @dataclass diff --git a/radiance/structures/colors.py b/pyrad/structures/colors.py similarity index 100% rename from radiance/structures/colors.py rename to pyrad/structures/colors.py diff --git a/radiance/structures/rays.py b/pyrad/structures/rays.py similarity index 99% rename from radiance/structures/rays.py rename to pyrad/structures/rays.py index 55b9875f8..f2458c527 100644 --- a/radiance/structures/rays.py +++ b/pyrad/structures/rays.py @@ -8,7 +8,7 @@ from typing import Optional import torch from torchtyping import TensorType -from radiance.utils.misc import is_not_none +from pyrad.utils.misc import is_not_none @dataclass diff --git a/radiance/utils/__init__.py b/pyrad/utils/__init__.py similarity index 100% rename from radiance/utils/__init__.py rename to pyrad/utils/__init__.py diff --git a/radiance/utils/callbacks.py b/pyrad/utils/callbacks.py similarity index 87% rename from radiance/utils/callbacks.py rename to pyrad/utils/callbacks.py index 8ce22cac0..578886104 100644 --- a/radiance/utils/callbacks.py +++ b/pyrad/utils/callbacks.py @@ -2,8 +2,8 @@ Callback functions for training iterations """ -from radiance.nerf.graph.base import Graph -from radiance.nerf.occupancy_grid import OccupancyGrid +from pyrad.nerf.graph.base import Graph +from pyrad.nerf.occupancy_grid import OccupancyGrid def update_occupancy(graph: Graph): diff --git a/radiance/utils/comms.py b/pyrad/utils/comms.py similarity index 100% rename from radiance/utils/comms.py rename to pyrad/utils/comms.py diff --git a/radiance/utils/decorators.py b/pyrad/utils/decorators.py similarity index 96% rename from radiance/utils/decorators.py rename to pyrad/utils/decorators.py index ca8be04a7..4765f1e4b 100644 --- a/radiance/utils/decorators.py +++ b/pyrad/utils/decorators.py @@ -3,7 +3,7 @@ Decorator definitions """ from typing import Callable, List -from radiance.utils import comms +from pyrad.utils import comms def decorate_all(decorators: List[Callable]) -> Callable: diff --git a/radiance/utils/io.py b/pyrad/utils/io.py similarity index 100% rename from radiance/utils/io.py rename to pyrad/utils/io.py diff --git a/radiance/utils/math.py b/pyrad/utils/math.py similarity index 100% rename from radiance/utils/math.py rename to pyrad/utils/math.py diff --git a/radiance/utils/misc.py b/pyrad/utils/misc.py similarity index 100% rename from radiance/utils/misc.py rename to pyrad/utils/misc.py diff --git a/radiance/utils/profiler.py b/pyrad/utils/profiler.py similarity index 94% rename from radiance/utils/profiler.py rename to pyrad/utils/profiler.py index 68dc2d29a..c21d1a59e 100644 --- a/radiance/utils/profiler.py +++ b/pyrad/utils/profiler.py @@ -7,8 +7,8 @@ from typing import Callable from omegaconf import DictConfig -from radiance.utils import comms -from radiance.utils.decorators import check_main_thread, check_profiler_enabled, decorate_all +from pyrad.utils import comms +from pyrad.utils.decorators import check_main_thread, check_profiler_enabled, decorate_all PROFILER = [] diff --git a/radiance/utils/visualization.py b/pyrad/utils/visualization.py similarity index 100% rename from radiance/utils/visualization.py rename to pyrad/utils/visualization.py diff --git a/radiance/utils/writer.py b/pyrad/utils/writer.py similarity index 98% rename from radiance/utils/writer.py rename to pyrad/utils/writer.py index 22a1dfbb7..fbf825ac6 100644 --- a/radiance/utils/writer.py +++ b/pyrad/utils/writer.py @@ -15,8 +15,8 @@ from omegaconf import DictConfig, ListConfig from torch.utils.tensorboard import SummaryWriter from torchtyping import TensorType -import radiance.utils.writer -from radiance.utils.decorators import check_main_thread, decorate_all +import pyrad.utils.writer +from pyrad.utils.decorators import check_main_thread, decorate_all to8b = lambda x: (255 * torch.clamp(x, min=0, max=1)).to(torch.uint8) EVENT_WRITERS = [] @@ -135,7 +135,7 @@ def setup_event_writers(config: DictConfig) -> None: """Initialization of all event writers specified in config""" logging_configs = config.logging.writer for writer_type in logging_configs: - writer_class = getattr(radiance.utils.writer, writer_type) + writer_class = getattr(pyrad.utils.writer, writer_type) writer_config = logging_configs[writer_type] if writer_type == "LocalWriter": curr_writer = writer_class(writer_config.save_dir, writer_config.stats_to_track, writer_config.max_log_size) diff --git a/radiance/viewer/README.md b/pyrad/viewer/README.md similarity index 100% rename from radiance/viewer/README.md rename to pyrad/viewer/README.md diff --git a/radiance/viewer/__init__.py b/pyrad/viewer/__init__.py similarity index 100% rename from radiance/viewer/__init__.py rename to pyrad/viewer/__init__.py diff --git a/radiance/viewer/plotly.py b/pyrad/viewer/plotly.py similarity index 98% rename from radiance/viewer/plotly.py rename to pyrad/viewer/plotly.py index 0964a5489..98a0128da 100644 --- a/radiance/viewer/plotly.py +++ b/pyrad/viewer/plotly.py @@ -6,7 +6,7 @@ import numpy as np import plotly.graph_objects as go import torch -from radiance.structures.rays import RayBundle +from pyrad.structures.rays import RayBundle def color_str(color): diff --git a/scripts/benchmarking/launch_train_blender.sh b/scripts/benchmarking/launch_train_blender.sh index 38bba313b..b834fe3c6 100755 --- a/scripts/benchmarking/launch_train_blender.sh +++ b/scripts/benchmarking/launch_train_blender.sh @@ -43,7 +43,7 @@ for dataset in ${DATASETS[@]}; do data.dataset.data_directory=data/blender/${dataset} \ data.dataset.downscale_factor=1 \ experiment_name=blender_${dataset}_${tag} \ - graph.model_dir=radiance_models/ \ + graph.model_dir=pyrad_models/ \ graph.steps_per_save=25000 \ graph.max_num_iterations=2000000 \ logging.enable_profiler=False & diff --git a/scripts/benchmarking/run_benchmark.py b/scripts/benchmarking/run_benchmark.py index 08912ea17..0e4e169ea 100644 --- a/scripts/benchmarking/run_benchmark.py +++ b/scripts/benchmarking/run_benchmark.py @@ -12,7 +12,7 @@ from hydra.core.global_hydra import GlobalHydra from omegaconf import DictConfig from tqdm import tqdm -from radiance.nerf.trainer import Trainer +from pyrad.nerf.trainer import Trainer BENCH = { "method": "vanilla_nerf", diff --git a/scripts/run_train.py b/scripts/run_train.py index e612c3547..fd6e70d43 100644 --- a/scripts/run_train.py +++ b/scripts/run_train.py @@ -18,8 +18,8 @@ import torch.multiprocessing as mp from omegaconf import DictConfig import yaml -from radiance.nerf.trainer import Trainer -from radiance.utils import comms, profiler +from pyrad.nerf.trainer import Trainer +from pyrad.utils import comms, profiler from time import time logging.basicConfig(format="[%(filename)s:%(lineno)d] %(message)s", level=logging.DEBUG) diff --git a/setup.cfg b/setup.cfg index 6565fd502..61aec33b9 100644 --- a/setup.cfg +++ b/setup.cfg @@ -1,7 +1,7 @@ [metadata] -name = radiance +name = pyrad version = 0.0.0 -url = https://github.com/ethanweber/radiance.git +url = https://github.com/ethanweber/pyrad.git description = differentiable volume rendering package long_description = file: README.md diff --git a/setup.py b/setup.py index 1f9a0460c..fe0bd2bb6 100644 --- a/setup.py +++ b/setup.py @@ -13,12 +13,12 @@ if USE_CUDA: ext_modules = [ CUDAExtension( - name="radiance_cuda", + name="pyrad_cuda", sources=[ - "radiance/cuda/radiance_cuda.cpp", - "radiance/cuda/radiance_cuda_kernel.cu", + "pyrad/cuda/pyrad_cuda.cpp", + "pyrad/cuda/pyrad_cuda_kernel.cu", ], - include_dirs=[os.path.join("radiance", "cuda", "include")], + include_dirs=[os.path.join("pyrad", "cuda", "include")], extra_compile_args={ "cxx": [], "nvcc": [], diff --git a/tests/cuda/test_mattport_cuda.py b/tests/cuda/test_mattport_cuda.py index f96f3d421..cb5efb5da 100644 --- a/tests/cuda/test_mattport_cuda.py +++ b/tests/cuda/test_mattport_cuda.py @@ -2,9 +2,9 @@ Test the CUDA bindings. """ import torch -from radiance.cuda.sampler import Sampler +from pyrad.cuda.sampler import Sampler import time -import radiance_cuda +import pyrad_cuda import os @@ -78,7 +78,7 @@ def test_benchmark(): print("PYTORCH_SAMPLE_TIME:", PYTORCH_SAMPLE_TIME) start = time.time() - time_steps_cu, samples_cu, time_steps_mask_cu = radiance_cuda.sample_uniformly_along_ray_bundle( + time_steps_cu, samples_cu, time_steps_mask_cu = pyrad_cuda.sample_uniformly_along_ray_bundle( origins, directions, nears, fars, offsets, max_num_samples ) CUDA_SAMPLE_TIME = time.time() - start diff --git a/tests/nerf/dataloader/test_ray_generator.py b/tests/nerf/dataloader/test_ray_generator.py index 0bd44e058..dad77d383 100644 --- a/tests/nerf/dataloader/test_ray_generator.py +++ b/tests/nerf/dataloader/test_ray_generator.py @@ -7,12 +7,12 @@ import torch from torch.utils.data import DataLoader from tqdm import tqdm -from radiance.nerf.dataset.collate import CollateIterDataset, collate_batch_size_one -from radiance.nerf.dataset.image_dataset import ImageDataset, collate_batch -from radiance.nerf.dataset.utils import get_dataset_inputs -from radiance.nerf.ray_generator import RayGenerator -from radiance.utils.io import get_absolute_path -from radiance.viewer.plotly import visualize_dataset +from pyrad.nerf.dataset.collate import CollateIterDataset, collate_batch_size_one +from pyrad.nerf.dataset.image_dataset import ImageDataset, collate_batch +from pyrad.nerf.dataset.utils import get_dataset_inputs +from pyrad.nerf.ray_generator import RayGenerator +from pyrad.utils.io import get_absolute_path +from pyrad.viewer.plotly import visualize_dataset def visualize_batch(batch): diff --git a/tests/nerf/field_modules/test_embedding.py b/tests/nerf/field_modules/test_embedding.py index 74c54c635..c8c682b77 100644 --- a/tests/nerf/field_modules/test_embedding.py +++ b/tests/nerf/field_modules/test_embedding.py @@ -1,7 +1,7 @@ """ Embedding tests """ -from radiance.nerf.field_modules.embedding import Embedding +from pyrad.nerf.field_modules.embedding import Embedding def test_indexing(): diff --git a/tests/nerf/field_modules/test_encodings.py b/tests/nerf/field_modules/test_encodings.py index 0c277f09d..921dc37f1 100644 --- a/tests/nerf/field_modules/test_encodings.py +++ b/tests/nerf/field_modules/test_encodings.py @@ -3,7 +3,7 @@ Encoding Tests """ import pytest import torch -from radiance.nerf.field_modules import encoding +from pyrad.nerf.field_modules import encoding def test_scaling_and_offset(): diff --git a/tests/nerf/field_modules/test_field_outputs.py b/tests/nerf/field_modules/test_field_outputs.py index 28ef9ea85..b7fe775ec 100644 --- a/tests/nerf/field_modules/test_field_outputs.py +++ b/tests/nerf/field_modules/test_field_outputs.py @@ -5,7 +5,7 @@ import torch from torch import nn import pytest -from radiance.nerf.field_modules.field_heads import ( +from pyrad.nerf.field_modules.field_heads import ( DensityFieldHead, FieldHead, FieldHeadNames, diff --git a/tests/nerf/field_modules/test_mlp.py b/tests/nerf/field_modules/test_mlp.py index e56b6a05a..25a071f7a 100644 --- a/tests/nerf/field_modules/test_mlp.py +++ b/tests/nerf/field_modules/test_mlp.py @@ -4,7 +4,7 @@ MLP Test import torch from torch import nn -from radiance.nerf.field_modules import MLP +from pyrad.nerf.field_modules import MLP def test_mlp(): diff --git a/tests/nerf/test_fields.py b/tests/nerf/test_fields.py index 427e12aa3..625407076 100644 --- a/tests/nerf/test_fields.py +++ b/tests/nerf/test_fields.py @@ -3,9 +3,9 @@ Test the fields """ import torch -from radiance.nerf.field_modules.field_heads import FieldHeadNames -from radiance.nerf.fields.instant_ngp_field import TCNNInstantNGPField -from radiance.structures.rays import PointSamples +from pyrad.nerf.field_modules.field_heads import FieldHeadNames +from pyrad.nerf.fields.instant_ngp_field import TCNNInstantNGPField +from pyrad.structures.rays import PointSamples def test_tcnn_instant_ngp_field(): diff --git a/tests/nerf/test_occupancy_grid.py b/tests/nerf/test_occupancy_grid.py index ab86982ea..4e10ed380 100644 --- a/tests/nerf/test_occupancy_grid.py +++ b/tests/nerf/test_occupancy_grid.py @@ -3,7 +3,7 @@ Code to test the occupancy grid. """ import torch -from radiance.nerf.occupancy_grid import OccupancyGrid +from pyrad.nerf.occupancy_grid import OccupancyGrid def test_occupancy_grid(): diff --git a/tests/nerf/test_renderers.py b/tests/nerf/test_renderers.py index bdd87f4c2..aa1d57c85 100644 --- a/tests/nerf/test_renderers.py +++ b/tests/nerf/test_renderers.py @@ -4,7 +4,7 @@ Test renderers import pytest import torch -from radiance.nerf import renderers +from pyrad.nerf import renderers def test_rgb_renderer(): diff --git a/tests/nerf/test_sampler.py b/tests/nerf/test_sampler.py index c97d763f0..819246d8f 100644 --- a/tests/nerf/test_sampler.py +++ b/tests/nerf/test_sampler.py @@ -2,10 +2,10 @@ Test samplers """ import torch -from radiance.nerf.sampler import PDFSampler, UniformSampler +from pyrad.nerf.sampler import PDFSampler, UniformSampler -from radiance.structures.rays import RayBundle -from radiance.nerf.colliders import NearFarCollider +from pyrad.structures.rays import RayBundle +from pyrad.nerf.colliders import NearFarCollider def test_uniform_sampler(): diff --git a/tests/structures/test_cameras.py b/tests/structures/test_cameras.py index cf8f9aeec..5cca2d1d5 100644 --- a/tests/structures/test_cameras.py +++ b/tests/structures/test_cameras.py @@ -4,8 +4,8 @@ Test the camera classes. import torch -from radiance.structures.cameras import PinholeCamera -from radiance.structures.rays import RayBundle +from pyrad.structures.cameras import PinholeCamera +from pyrad.structures.rays import RayBundle def test_pinhole_camera():