run_dataset

Attributes

URL
log

Functions

args(→ argparse.Namespace)
download(→ None)
ensure_data(→ list[pathlib.Path])
images_in(→ list[pathlib.Path])
load(→ list[torch.Tensor])
novel_inputs(→ tuple[dict[str, torch.Tensor], ...)
main(→ None)

Module Contents

run_dataset.URL = 'http://storage.googleapis.com/gresearch/refraw360/360_v2.zip'

run_dataset.log

run_dataset.args() → argparse.Namespace

async run_dataset.download(url: str, dst: pathlib.Path) → None

run_dataset.ensure_data(root: pathlib.Path, scene: str) → list[pathlib.Path]

run_dataset.images_in(folder: pathlib.Path) → list[pathlib.Path]

run_dataset.load(paths: list[pathlib.Path], size: int) → list[torch.Tensor]

run_dataset.novel_inputs(images: list[torch.Tensor]) → tuple[dict[str, torch.Tensor], dict[str, int]]

run_dataset.main() → None