pe.data.image.camelyon17 module

class pe.data.image.camelyon17.Camelyon17(split='train', root_dir='data', res=64)[source]

Bases: Data

The Camelyon17 dataset.

__init__(split='train', root_dir='data', res=64)[source]

Constructor.

Parameters:
  • split (str, optional) – The split of the dataset. It should be either “train”, “val”, or “test”, defaults to “train”

  • root_dir (str, optional) – The root directory to save the dataset, defaults to “data”

  • res (int, optional) – The resolution of the images, defaults to 64

Raises:

ValueError – If the split is invalid