edflow.datasets.celeba module¶

Summary¶

Classes:

`CelebA`
`CelebATest`
`CelebATrain`
`CelebAVal`

Reference¶

class edflow.datasets.celeba.CelebA(config=None)[source]¶

Bases: edflow.data.dataset_mixin.DatasetMixin

NAME = 'CelebA'¶

URL = 'http://mmlab.ie.cuhk.edu.hk/projects/CelebA.html'¶

FILES = ['img_align_celeba.zip', 'list_eval_partition.txt', 'identity_CelebA.txt', 'list_attr_celeba.txt']¶

__init__(config=None)[source]¶: Initialize self. See help(type(self)) for accurate signature.

get_example(i)[source]¶

Note

Please the documentation of DatasetMixin to not be confused.

Add default behaviour for datasets defining an attribute data, which in turn is a dataset. This happens often when stacking several datasets on top of each other.

The default behaviour now is to return self.data.get_example(idx) if possible, and otherwise revert to the original behaviour.

class edflow.datasets.celeba.CelebATrain(config=None)[source]¶: Bases: edflow.datasets.celeba.CelebA

class edflow.datasets.celeba.CelebAVal(config=None)[source]¶: Bases: edflow.datasets.celeba.CelebA

class edflow.datasets.celeba.CelebATest(config=None)[source]¶: Bases: edflow.datasets.celeba.CelebA