data.loader package

Submodules

data.loader.dataloader module

class data.loader.dataloader.CVNetsDataLoader(dataset: BaseDataset, batch_size: int, batch_sampler: Sampler, num_workers: int | None = 1, pin_memory: bool | None = False, persistent_workers: bool | None = False, collate_fn: Optional = None, prefetch_factor: int | None = 2, *args, **kwargs)

Bases: DataLoader

This class extends PyTorch’s Dataloader

__init__(dataset: BaseDataset, batch_size: int, batch_sampler: Sampler, num_workers: int | None = 1, pin_memory: bool | None = False, persistent_workers: bool | None = False, collate_fn: Optional = None, prefetch_factor: int | None = 2, *args, **kwargs)

update_indices(new_indices: List, *args, **kwargs)

Update indices in the dataset class

samples_in_dataset()

Number of samples in the dataset

get_sample_indices() → List

Sample IDs

Module contents