trw.datasets.utils

Module Contents

Functions

pic_to_tensor(pic: PIL.Image) → torch.Tensor
pic_to_numpy(pic: PIL.Image) → numpy.ndarray
get_data_root(data_root: Optional[str]) → str	Returns the location where all the data will be stored.
download_and_extract_archive(url: str, dataset_path: str) → None

trw.datasets.utils.pic_to_tensor(pic: PIL.Image) → torch.Tensor

trw.datasets.utils.pic_to_numpy(pic: PIL.Image) → numpy.ndarray

trw.datasets.utils.get_data_root(data_root: Optional[str]) → str

Returns the location where all the data will be stored.

data_root: a path where to store the data. if data_root is None,

the environment variable TRW_DATA_ROOT will be used. If it is not defined, the default location of get_logging_root will be used instead.

trw.datasets.utils.download_and_extract_archive(url: str, dataset_path: str) → None