INNER CODE UNIT · Python

make_test

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

def make_test(name: str, config):
    match name:
        case "nerf":
            dataset = NeRFDataset(
                config.path,
                split="test",
                bg_color=config.model.background.color,
            )
        case "colmap":
            gsplat_image_downscale = config.dataset.get("gsplat_image_downscale", False)
            # Load EXIF exposure data if enabled
            if config.dataset.get("load_exif", True):
                exif_exposures = _load_colmap_exif_exposures(
                    config.path,
                    config.dataset.downsample_factor,
                    original_images=gsplat_image_downscale,
                )
            else:

View source record →