mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-13 11:24:00 +08:00
Ethan Weber
457788d689
* renaming dataloader to data. adding some dataloader classes * restructuring configs * fixed instant ngp tcnn direction rescaling * working through test cases * black and lint and tests
22 lines
820 B
Python
22 lines
820 B
Python
"""
|
|
Callback functions for training iterations
|
|
"""
|
|
|
|
from pyrad.graphs.base import Graph
|
|
from pyrad.fields.occupancy_fields.occupancy_grid import OccupancyGrid
|
|
|
|
|
|
def update_occupancy(graph: Graph):
|
|
"""Update the occupancy grid depending on iteration count after every train step"""
|
|
if not hasattr(graph, "occupancy_pointers"):
|
|
occupancy_grids = []
|
|
for _, module in graph.__dict__["_modules"].items():
|
|
if isinstance(module, OccupancyGrid):
|
|
occupancy_grids.append(module)
|
|
graph.occupancy_pointers = occupancy_grids
|
|
for grid in graph.occupancy_pointers:
|
|
grid.iteration_count += 1
|
|
if grid.iteration_count % grid.update_every_num_iters == 0:
|
|
if grid.density_fn is not None:
|
|
grid.update_occupancy_grid(grid.density_fn)
|