mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-13 03:14: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
79 lines
1.9 KiB
Python
79 lines
1.9 KiB
Python
"""
|
|
Test renderers
|
|
"""
|
|
import pytest
|
|
import torch
|
|
|
|
from pyrad.renderers import renderers
|
|
|
|
|
|
def test_rgb_renderer():
|
|
"""Test RGB volumetric rendering"""
|
|
num_samples = 10
|
|
|
|
rgb_samples = torch.ones((3, num_samples, 3))
|
|
weights = torch.ones((3, num_samples))
|
|
weights /= torch.sum(weights, axis=-1, keepdim=True)
|
|
|
|
rgb_renderer = renderers.RGBRenderer()
|
|
|
|
rgb = rgb_renderer(rgb=rgb_samples, weights=weights)
|
|
assert torch.max(rgb) > 0.9
|
|
|
|
rgb = rgb_renderer(rgb=rgb_samples * 0, weights=weights)
|
|
assert torch.max(rgb) == pytest.approx(0, abs=1e-6)
|
|
|
|
|
|
def test_sh_renderer():
|
|
"""Test SH volumetric rendering"""
|
|
|
|
levels = 4
|
|
num_samples = 10
|
|
|
|
sh = torch.ones((3, num_samples, 3 * levels**2))
|
|
weights = torch.ones((3, num_samples))
|
|
weights /= torch.sum(weights, axis=-1, keepdim=True)
|
|
directions = torch.zeros((3, num_samples, 3))
|
|
directions[..., 0] = 1
|
|
|
|
sh_renderer = renderers.SHRenderer()
|
|
|
|
rgb = sh_renderer(sh=sh, directions=directions, weights=weights)
|
|
assert torch.max(rgb) > 0.9
|
|
|
|
|
|
def test_acc_renderer():
|
|
"""Test accumulation rendering"""
|
|
|
|
num_samples = 10
|
|
weights = torch.ones((3, num_samples))
|
|
weights /= torch.sum(weights, axis=-1, keepdim=True)
|
|
|
|
acc_renderer = renderers.AccumulationRenderer()
|
|
|
|
accumulation = acc_renderer(weights=weights)
|
|
assert torch.max(accumulation) > 0.9
|
|
|
|
|
|
def test_depth_renderer():
|
|
"""Test depth rendering"""
|
|
|
|
num_samples = 10
|
|
weights = torch.ones((3, num_samples))
|
|
weights /= torch.sum(weights, axis=-1, keepdim=True)
|
|
|
|
bins = torch.linspace(0, 100, num_samples + 1)
|
|
bins = torch.stack([bins, bins, bins], dim=0)
|
|
|
|
depth_renderer = renderers.DepthRenderer()
|
|
|
|
depth = depth_renderer(weights=weights, bins=bins)
|
|
assert torch.min(depth) > 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_rgb_renderer()
|
|
test_sh_renderer()
|
|
test_acc_renderer()
|
|
test_depth_renderer()
|