benzina.torch.dataset

class benzina.torch.dataset.Dataset(archive: Union[str, benzina.utils.file.Track] = None, track: Union[str, benzina.utils.file.Track] = 'bzna_input')[source]
Parameters:
  • archive (str or Track) – path to the archive or a Track. If a Track, track will be ignored.
  • track (str or Track, optional) – track label or a Track. If a Track, archive must not be specified. (default: "bzna_input")
class benzina.torch.dataset.ClassificationDataset(archive: Union[str, Tuple[Union[str, benzina.utils.file.Track], Union[str, benzina.utils.file.Track]]] = None, tracks: Tuple[Union[str, benzina.utils.file.Track], Union[str, benzina.utils.file.Track]] = ('bzna_input', 'bzna_target'), input_label: str = 'bzna_thumb')[source]
Parameters:
  • archive (str or pair of Track) – path to the archive or a pair of Track. If a pair of Track, tracks will be ignored.
  • tracks (pair of str or Track, optional) – pair of input and target tracks labels or a pair of input and target Track. If a pair of Track, archive must not be specified. (default: ("bzna_input", "bzna_target"))
  • input_label (str, optional) – label of the inputs to use in the input track. (default: "bzna_thumb")
class benzina.torch.dataset.ImageNet(root: Union[str, Tuple[Union[str, benzina.utils.file.Track], Union[str, benzina.utils.file.Track]]] = None, split: str = None, tracks: Tuple[Union[str, benzina.utils.file.Track], Union[str, benzina.utils.file.Track]] = ('bzna_input', 'bzna_target'), input_label: str = 'bzna_thumb')[source]
Parameters:
  • root (str or pair of Track) – root of the ImageNet dataset or path to the archive or a pair of Track. If a pair of Track, tracks will be ignored.
  • split (None or str, optional) – The dataset split, supports test, train, val. If not specified, samples will be drawn from all splits.
  • tracks (pair of str or Track, optional) – pair of input and target tracks labels or a pair of input and target Track. If a pair of Track, root must not be specified. (default: ("bzna_input", "bzna_target"))
  • input_label (str, optional) – label of the inputs to use in the input track. (default: "bzna_thumb")