mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-11 02:24:38 +08:00
e38040ee4d
* working data loader code * adding gitignore * fixing linting errors * set black line length in settings file * update * fixed filepath * black format * tqdm
17 lines
516 B
Python
17 lines
516 B
Python
"""
|
|
Test the camera classes.
|
|
"""
|
|
|
|
import torch
|
|
|
|
from mattport.structures.cameras import PinholeCamera
|
|
from mattport.structures.rays import CameraRayBundle
|
|
|
|
|
|
def test_pinhole_camera():
|
|
"""Test that the pinhole camera model works."""
|
|
pinhole_camera = PinholeCamera(cx=400.0, cy=400.0, fx=10.0, fy=10.0)
|
|
camera_ray_bundle = pinhole_camera.generate_camera_rays()
|
|
assert isinstance(camera_ray_bundle, CameraRayBundle)
|
|
assert torch.allclose(camera_ray_bundle.origins[0], torch.tensor([0.0, 0.0, 0.0]))
|