basicsr.data.ffhq_dataset

class basicsr.data.ffhq_dataset.FFHQDataset(opt)

Bases: Dataset

FFHQ dataset for StyleGAN.

Parameters:

opt (dict) – Config for train datasets. It contains the following keys: dataroot_gt (str): Data root path for gt. io_backend (dict): IO backend type and other kwarg. mean (list | tuple): Image mean. std (list | tuple): Image std. use_hflip (bool): Whether to horizontally flip.