clean code
This commit is contained in:
parent
403cba76cf
commit
2fa17e9659
1 changed files with 2 additions and 4 deletions
|
@ -149,8 +149,7 @@ def render_single_image(rank, world_size, models, ray_sampler, chunk_size):
|
||||||
if m == 0:
|
if m == 0:
|
||||||
# foreground depth
|
# foreground depth
|
||||||
fg_far_depth = intersect_sphere(ray_o, ray_d) # [...,]
|
fg_far_depth = intersect_sphere(ray_o, ray_d) # [...,]
|
||||||
# fg_near_depth = 0.18 * torch.ones_like(fg_far_depth)
|
fg_near_depth = min_depth # [..., ]
|
||||||
fg_near_depth = min_depth # [..., 3]
|
|
||||||
step = (fg_far_depth - fg_near_depth) / (N_samples - 1)
|
step = (fg_far_depth - fg_near_depth) / (N_samples - 1)
|
||||||
fg_depth = torch.stack([fg_near_depth + i * step for i in range(N_samples)], dim=-1) # [..., N_samples]
|
fg_depth = torch.stack([fg_near_depth + i * step for i in range(N_samples)], dim=-1) # [..., N_samples]
|
||||||
|
|
||||||
|
@ -413,8 +412,7 @@ def ddp_train_nerf(rank, args):
|
||||||
if m == 0:
|
if m == 0:
|
||||||
# foreground depth
|
# foreground depth
|
||||||
fg_far_depth = intersect_sphere(ray_batch['ray_o'], ray_batch['ray_d']) # [...,]
|
fg_far_depth = intersect_sphere(ray_batch['ray_o'], ray_batch['ray_d']) # [...,]
|
||||||
# fg_near_depth = 0.18 * torch.ones_like(fg_far_depth)
|
fg_near_depth = ray_batch['min_depth'] # [..., ]
|
||||||
fg_near_depth = ray_batch['min_depth'] # [..., 3]
|
|
||||||
step = (fg_far_depth - fg_near_depth) / (N_samples - 1)
|
step = (fg_far_depth - fg_near_depth) / (N_samples - 1)
|
||||||
fg_depth = torch.stack([fg_near_depth + i * step for i in range(N_samples)], dim=-1) # [..., N_samples]
|
fg_depth = torch.stack([fg_near_depth + i * step for i in range(N_samples)], dim=-1) # [..., N_samples]
|
||||||
fg_depth = perturb_samples(fg_depth) # random perturbation during training
|
fg_depth = perturb_samples(fg_depth) # random perturbation during training
|
||||||
|
|
Loading…
Reference in a new issue