robustness.tools.folder module

robustness.tools.folder.has_file_allowed_extension(filename, extensions)

Checks if a file is an allowed extension.

Parameters:
  • filename (string) – path to a file
  • extensions (iterable of strings) – extensions to consider (lowercase)
Returns:

True if the filename ends with one of given extensions
Return type:

bool
robustness.tools.folder.is_image_file(filename)

Checks if a file is an allowed image extension.

Parameters:filename (string) – path to a file
Returns:True if the filename ends with a known image extension
Return type:bool
robustness.tools.folder.make_dataset(dir, class_to_idx, extensions)
class robustness.tools.folder.DatasetFolder(root, loader, extensions, transform=None, target_transform=None, label_mapping=None)

Bases: sphinx.ext.autodoc.importer._MockObject

A generic data loader where the samples are arranged in this way:

root/class_x/xxx.ext
root/class_x/xxy.ext
root/class_x/xxz.ext

root/class_y/123.ext
root/class_y/nsdf3.ext
root/class_y/asd932_.ext
Parameters:
  • root (string) – Root directory path.
  • loader (callable) – A function to load a sample given its path.
  • extensions (list[string]) – A list of allowed extensions.
  • transform (callable, optional) – A function/transform that takes in a sample and returns a transformed version. E.g, transforms.RandomCrop for images.
  • target_transform – A function/transform that takes in the target and transforms it.
robustness.tools.folder.pil_loader(path)
robustness.tools.folder.accimage_loader(path)
robustness.tools.folder.default_loader(path)
class robustness.tools.folder.ImageFolder(root, transform=None, target_transform=None, loader=<function default_loader>, label_mapping=None)

Bases: robustness.tools.folder.DatasetFolder

A generic data loader where the images are arranged in this way:

root/dog/xxx.png
root/dog/xxy.png
root/dog/xxz.png

root/cat/123.png
root/cat/nsdf3.png
root/cat/asd932_.png
Parameters:
  • root (string) – Root directory path.
  • transform (callable, optional) – A function/transform that takes in an PIL image and returns a transformed version. E.g, transforms.RandomCrop
  • target_transform (callable, optional) – A function/transform that takes in the target and transforms it.
  • loader – A function to load an image given its path.
class robustness.tools.folder.TensorDataset(*tensors, transform=None)

Bases: sphinx.ext.autodoc.importer._MockObject

Dataset wrapping tensors.

Each sample will be retrieved by indexing tensors along the first dimension.

Parameters:*tensors (Tensor) – tensors that have the same size of the first dimension.