steganogan.loader module

class steganogan.loader.DataLoader(path, transform=None, limit=inf, shuffle=True, num_workers=8, batch_size=4, *args, **kwargs)

Bases: torch.utils.data.dataloader.DataLoader

class steganogan.loader.ImageFolder(path, transform, limit=inf)

Bases: torchvision.datasets.folder.ImageFolder