run_dataset
===========

.. py:module:: run_dataset


Attributes
----------

.. autoapisummary::

   run_dataset.URL
   run_dataset.log


Functions
---------

.. autoapisummary::

   run_dataset.args
   run_dataset.download
   run_dataset.ensure_data
   run_dataset.images_in
   run_dataset.load
   run_dataset.novel_inputs
   run_dataset.main


Module Contents
---------------

.. py:data:: URL
   :value: 'http://storage.googleapis.com/gresearch/refraw360/360_v2.zip'


.. py:data:: log

.. py:function:: args() -> argparse.Namespace

.. py:function:: download(url: str, dst: pathlib.Path) -> None
   :async:


.. py:function:: ensure_data(root: pathlib.Path, scene: str) -> list[pathlib.Path]

.. py:function:: images_in(folder: pathlib.Path) -> list[pathlib.Path]

.. py:function:: load(paths: list[pathlib.Path], size: int) -> list[torch.Tensor]

.. py:function:: novel_inputs(images: list[torch.Tensor]) -> tuple[dict[str, torch.Tensor], dict[str, int]]

.. py:function:: main() -> None