nerfbaselines.datasets

class nerfbaselines.datasets.Dataset(cameras: nerfbaselines.cameras.Cameras, file_paths: List[str], sampling_mask_paths: Optional[List[str]] = None, file_paths_root: Optional[pathlib.Path] = None, images: Optional[numpy.ndarray] = None, sampling_masks: Optional[numpy.ndarray] = None, points3D_xyz: Optional[numpy.ndarray] = None, points3D_rgb: Optional[numpy.ndarray] = None, metadata: Dict = <factory>, color_space: Optional[Literal['srgb', 'linear']] = None)

Bases: object

cameras: Cameras

color_space: Literal['srgb', 'linear'] | None = None

property expected_scene_scale

file_paths: List[str]

file_paths_root: Path | None = None

images: ndarray | None = None

load_features(required_features)

metadata: Dict

points3D_rgb: ndarray | None = None

points3D_xyz: ndarray | None = None

sampling_mask_paths: List[str] | None = None

sampling_masks: ndarray | None = None

exception nerfbaselines.datasets.DatasetNotFoundError

Bases: Exception

nerfbaselines.datasets.load_dataset(path: Path, split: str, features: FrozenSet[Literal['color', 'points3D_xyz']]) → Dataset

nerfbaselines.datasets.colmap

class nerfbaselines.datasets.colmap.Camera(id, model, width, height, params)

Bases: tuple

class nerfbaselines.datasets.colmap.Image(id, qvec, tvec, camera_id, name, xys, point3D_ids)

Bases: BaseImage

qvec2rotmat()

class nerfbaselines.datasets.colmap.Point3D(id, xyz, rgb, error, image_ids, point2D_idxs)

Bases: tuple

nerfbaselines.datasets.colmap.load_colmap_dataset(path: Path, images_path: Path | None = None, split: str | None = None, test_indices: Indices | None = None, features: FrozenSet[Literal['color', 'points3D_xyz']] | None = None)

nerfbaselines.datasets.colmap.padded_stack(tensors: List[ndarray]) → ndarray

nerfbaselines.datasets.colmap.qvec2rotmat(qvec)

nerfbaselines.datasets.colmap.read_cameras_binary(path_to_model_file)

see: src/base/reconstruction.cc

void Reconstruction::WriteCamerasBinary(const std::string& path) void Reconstruction::ReadCamerasBinary(const std::string& path)

nerfbaselines.datasets.colmap.read_cameras_text(path)

see: src/base/reconstruction.cc

void Reconstruction::WriteCamerasText(const std::string& path) void Reconstruction::ReadCamerasText(const std::string& path)

nerfbaselines.datasets.colmap.read_images_binary(path_to_model_file)

see: src/base/reconstruction.cc

void Reconstruction::ReadImagesBinary(const std::string& path) void Reconstruction::WriteImagesBinary(const std::string& path)

nerfbaselines.datasets.colmap.read_images_text(path)

see: src/base/reconstruction.cc

void Reconstruction::ReadImagesText(const std::string& path) void Reconstruction::WriteImagesText(const std::string& path)

nerfbaselines.datasets.colmap.read_points3D_binary(path_to_model_file)

see: src/base/reconstruction.cc

void Reconstruction::ReadPoints3DBinary(const std::string& path) void Reconstruction::WritePoints3DBinary(const std::string& path)

nerfbaselines.datasets.colmap.read_points3D_text(path)

see: src/base/reconstruction.cc

void Reconstruction::ReadPoints3DText(const std::string& path) void Reconstruction::WritePoints3DText(const std::string& path)

nerfbaselines.datasets.mipnerf360

nerfbaselines.datasets.mipnerf360.load_colmap_dataset(path: Path, images_path: Path | None = None, split: str | None = None, test_indices: Indices | None = None, features: FrozenSet[Literal['color', 'points3D_xyz']] | None = None)

nerfbaselines.datasets.mipnerf360.load_mipnerf360_dataset(path: Path, split: str, **kwargs)

nerfbaselines.datasets.mipnerf360.single(xs)