mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-02 03:42:50 +08:00
fba419b7f7
Summary: An initial NeRF diff which sets up the folder structure and implements the raymarching algorithm of NeRF. Reviewed By: nikhilaravi Differential Revision: D25623990 fbshipit-source-id: ac6b05a9b866358bd4bbf44858f06859d8a6ebd1
35 lines
964 B
Python
35 lines
964 B
Python
# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
|
|
|
|
import unittest
|
|
|
|
import torch
|
|
from nerf.raymarcher import EmissionAbsorptionNeRFRaymarcher
|
|
from pytorch3d.renderer import EmissionAbsorptionRaymarcher
|
|
|
|
|
|
class TestRaymarcher(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
torch.manual_seed(42)
|
|
|
|
def test_raymarcher(self):
|
|
"""
|
|
Checks that the nerf raymarcher outputs are identical to the
|
|
EmissionAbsorptionRaymarcher.
|
|
"""
|
|
|
|
feat_dim = 3
|
|
rays_densities = torch.rand(100, 10, 1)
|
|
rays_features = torch.randn(100, 10, feat_dim)
|
|
|
|
out, out_nerf = [
|
|
raymarcher(rays_densities, rays_features)
|
|
for raymarcher in (
|
|
EmissionAbsorptionRaymarcher(),
|
|
EmissionAbsorptionNeRFRaymarcher(),
|
|
)
|
|
]
|
|
|
|
self.assertTrue(
|
|
torch.allclose(out[..., :feat_dim], out_nerf[0][..., :feat_dim])
|
|
)
|