nntoolbox.vision.utils.dataset module

class nntoolbox.vision.utils.dataset.FaceScrub(root, data_path, transform=None)

Bases: Generic[torch.utils.data.dataset.T_co]

nntoolbox.vision.utils.dataset.download_facescrub(root: str, data_path: str, max_size: int = 128)