INNER CODE UNIT · Python

make

nv-tlabs/3dgrut · threedgrut/datasets/__init__.py:68

def make(name: str, config, ray_jitter):
    match name:
        case "nerf":
            train_dataset = NeRFDataset(
                config.path,
                split="train",
                bg_color=config.model.background.color,
                ray_jitter=ray_jitter,
            )
            val_dataset = NeRFDataset(
                config.path,
                split="val",
                bg_color=config.model.background.color,
            )
        case "colmap":
            gsplat_image_downscale = config.dataset.get("gsplat_image_downscale", False)
            # Load EXIF exposure data if enabled (shared between train and val)
            if config.dataset.get("load_exif", True):

View source record →