otthorn/nerf_plus_plus
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

throw exception if cameras are not normalized properly

Browse source
This commit is contained in:
Kai Zhang 2020-11-08 12:16:18 -05:00 committed by GitHub
parent acc089a449
commit 5792048384
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
ddp_train_nerf.py
View file

@ -49,7 +49,10 @@ def intersect_sphere(ray_o, ray_d):
p = ray_o + d1.unsqueeze(-1) * ray_d p = ray_o + d1.unsqueeze(-1) * ray_d
# consider the case where the ray does not intersect the sphere # consider the case where the ray does not intersect the sphere
ray_d_cos = 1. / torch.norm(ray_d, dim=-1) ray_d_cos = 1. / torch.norm(ray_d, dim=-1)
d2 = torch.sqrt(1. - torch.sum(p * p, dim=-1)) * ray_d_cos p_norm_sq = torch.sum(p * p, dim=-1)
if (p_norm_sq >= 1.).any():
raise Exception('Not all your cameras are bounded by the unit sphere; please make sure the cameras are normalized properly!')
d2 = torch.sqrt(1. - p_norm_sq) * ray_d_cos
return d1 + d2 return d1 + d2