nerfbaselines¶
- class nerfbaselines.CameraModel(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)[source]¶
Bases:
Enum
- FULL_OPENCV = 3¶
- OPENCV = 1¶
- OPENCV_FISHEYE = 2¶
- PINHOLE = 0¶
- class nerfbaselines.Cameras(poses: numpy.ndarray, normalized_intrinsics: numpy.ndarray, camera_types: numpy.ndarray, distortion_parameters: numpy.ndarray, image_sizes: numpy.ndarray | None, nears_fars: numpy.ndarray | None)[source]¶
Bases:
object
- camera_types: ndarray¶
- distortion_parameters: ndarray¶
- get_rays(xy: ~numpy.ndarray, xnp=<module 'numpy' from '/opt/hostedtoolcache/Python/3.12.6/x64/lib/python3.12/site-packages/numpy/__init__.py'>) Tuple[ndarray, ndarray] [source]¶
- image_sizes: ndarray | None¶
- nears_fars: ndarray | None¶
- normalized_intrinsics: ndarray¶
- poses: ndarray¶
- project(xyz: ~numpy.ndarray, xnp=<module 'numpy' from '/opt/hostedtoolcache/Python/3.12.6/x64/lib/python3.12/site-packages/numpy/__init__.py'>) ndarray [source]¶
- class nerfbaselines.CurrentProgress(i: int, total: int, image_i: int, image_total: int)[source]¶
Bases:
object
- i: int¶
- image_i: int¶
- image_total: int¶
- total: int¶
- class nerfbaselines.Method(*args, **kwargs)[source]¶
Bases:
Protocol
- abstract get_info() MethodInfo [source]¶
Get method defaults for the trainer.
- Returns:
Method info.
- abstract render(cameras: Cameras, progress_callback: Callable[[CurrentProgress], None] | None = None) Iterable[Dict] [source]¶
Render images.
- Parameters:
cameras – Cameras.
progress_callback – Callback for progress.
- class nerfbaselines.MethodInfo(loaded_step: Optional[int] = None, num_iterations: Optional[int] = None, batch_size: Optional[int] = None, eval_batch_size: Optional[int] = None, required_features: FrozenSet[Literal['color', 'points3D_xyz', 'points3D_rgb']] = <factory>, supported_camera_models: FrozenSet = <factory>)[source]¶
Bases:
object
- batch_size: int | None = None¶
- eval_batch_size: int | None = None¶
- loaded_step: int | None = None¶
- num_iterations: int | None = None¶
- required_features: FrozenSet[Literal['color', 'points3D_xyz', 'points3D_rgb']]¶
- supported_camera_models: FrozenSet¶
- class nerfbaselines.RayMethod(batch_size, seed: int = 42, xnp=<module 'numpy' from '/opt/hostedtoolcache/Python/3.12.6/x64/lib/python3.12/site-packages/numpy/__init__.py'>)[source]¶
Bases:
Method
- render(cameras: Cameras, progress_callback: Callable[[CurrentProgress], None] | None = None) Iterable[Dict] [source]¶
Render images.
- Parameters:
cameras – Cameras.
progress_callback – Callback for progress.
- abstract render_rays(origins: ndarray, directions: ndarray, nears_fars: ndarray | None) Dict [source]¶
Render rays.
- Parameters:
origins – Ray origins.
directions – Ray directions.
nears_fars – Near and far planes.