tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 02:24:38 +08:00
Code

instant ngp fix (#80)

Browse Source
This commit is contained in:
Ethan Weber 2022-07-05 14:36:35 -07:00 committed by GitHub
parent ced02a00be
commit e507c97e3e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pyrad/engine/trainer.py
View File

@ -177,8 +177,8 @@ class Trainer:
self.optimizers.optimizer_step_all()
self.optimizers.scheduler_step_all(step)
if self.graph.callbacks:
for _func in self.graph.callbacks:
_func.after_step(step)
for func_ in self.graph.callbacks:
func_.after_step(step)
return loss_dict
@profiler.time_function

5
pyrad/graphs/instant_ngp.py
View File

@ -105,7 +105,10 @@ class NGPGraph(Graph):
ts_uniform = (ray_samples_uniform.bin_starts + ray_samples_uniform.bin_ends) / 2.0
ts_pdf = (ray_samples_pdf.bin_starts + ray_samples_pdf.bin_ends) / 2.0
ts, indices = torch.sort(torch.cat([ts_uniform, ts_pdf], -1), -1)
ray_samples = ray_bundle.get_ray_samples(bin_starts=ts, bin_ends=ts)
bin_starts = ts
bin_ends = torch.cat([ts[..., 1:], ts[..., -1:]], dim=-1)
ray_samples = ray_bundle.get_ray_samples(bin_starts=bin_starts, bin_ends=bin_ends)
field_outputs = {}
for fo_name, _ in field_outputs_pdf.items():
fo_uniform = field_outputs_uniform[fo_name]