Methodcal_loss_rgb(self, geometry, euler, trans, cam, ori_img, light, texture, lands)
data_utils/face_tracking/render_land.py:172
Methodforward(ctx, inputs, embeddings, offsets, per_level_scale, base_resolution, calc_grad_inputs=False, gridtype=0, align
gridencoder/grid.py:27
Methodforward(
ctx, proj_geo, texture, nbl, ori_img, is_visible, tri_inds, pixel_valid
)
data_utils/face_tracking/render_land.py:66
Methodforward(
self, proj_geo, texture, nbl, ori_img, is_visible, tri_inds, pixel_valid
)
data_utils/face_tracking/render_land.py:119
Methodforward morton3D, CUDA implementation Args: coords: [N, 3], int32, in [0, 128) (for some reason there is no uint32 tensor in torch...)
raymarching/raymarching.py:84
Methodforward morton3D_invert, CUDA implementation Args: indices: [N], int32, in [0, 128^3) Returns: coords: [N, 3], int32,
raymarching/raymarching.py:107
Methodforward composite rays' rgbs, according to the ray marching formula. Args: rgbs: float, [M, 3] sigmas: float, [M,]
raymarching/raymarching.py:286
Methodrender(self, rays_o, rays_d, auds, bg_coords, poses, staged=False, max_ray_batch=4096, **kwargs)
nerf/renderer.py:504
Methodrender_mesh(self, geometry, euler, trans, cam, ori_img, light)
data_utils/face_tracking/render_land.py:147
Methodrun_cuda(self, rays_o, rays_d, auds, bg_coords, poses, eye=None, index=0, dt_gamma=0, bg_color=None, perturb=False, fo
nerf/renderer.py:158