NYUv2¶

class torch_uncertainty.datasets.NYUv2(root, split, transforms=None, min_depth=0.0, max_depth=10.0, download=False)[source]¶

NYUv2 depth dataset.

Parameters:

root (Path | str) – Root directory where dataset is stored.
split (Literal["train", "val"]) – Dataset split.
transforms (Callable | None) – Transform to apply to samples & targets. Defaults to None.
min_depth (float) – Minimum depth value. Defaults to 1e-3.
max_depth (float) – Maximum depth value. Defaults to 10.
download (bool) – Download dataset if not found. Defaults to False.