Skip to content

Commit

Permalink
Fixing Assertion issues when exporting NeRFs to poisson meshes and po…
Browse files Browse the repository at this point in the history
…intclouds (#3587)

fixing assertion issues for a successful poissonMesh and pointcloud export
  • Loading branch information
AntonioMacaronio authored Feb 10, 2025
1 parent 4a3e3e6 commit f604fd9
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions nerfstudio/scripts/exporter.py
Original file line number Diff line number Diff line change
Expand Up @@ -141,8 +141,9 @@ def main(self) -> None:
pipeline.datamanager,
(VanillaDataManager, ParallelDataManager),
)
assert pipeline.datamanager.train_pixel_sampler is not None
pipeline.datamanager.train_pixel_sampler.num_rays_per_batch = self.num_rays_per_batch
if isinstance(pipeline.datamanager, VanillaDataManager):
assert pipeline.datamanager.train_pixel_sampler is not None
pipeline.datamanager.train_pixel_sampler.num_rays_per_batch = self.num_rays_per_batch

# Whether the normals should be estimated based on the point cloud.
estimate_normals = self.normal_method == "open3d"
Expand Down Expand Up @@ -329,8 +330,9 @@ def main(self) -> None:
pipeline.datamanager,
(VanillaDataManager, ParallelDataManager),
)
assert pipeline.datamanager.train_pixel_sampler is not None
pipeline.datamanager.train_pixel_sampler.num_rays_per_batch = self.num_rays_per_batch
if isinstance(pipeline.datamanager, VanillaDataManager):
assert pipeline.datamanager.train_pixel_sampler is not None
pipeline.datamanager.train_pixel_sampler.num_rays_per_batch = self.num_rays_per_batch

# Whether the normals should be estimated based on the point cloud.
estimate_normals = self.normal_method == "open3d"
Expand Down

0 comments on commit f604fd9

Please sign in to comment.