def __init__(self, size): super().__init__() d1 = FFHQValidation(size=size, keys=["image"]) d2 = OilPortraitsValidation(size=size, keys=["image"]) self.data = ConcatDatasetWithIndex([d1, d2])
def __init__(self, size): super().__init__() d1 = ConcatDataset([FFHQTrain(size=size, keys=["image"]), CelebAHQTrain(size=size, keys=["image"])]) d2 = OilPortraitsTrain(size=size, keys=["image"]) self.data = ConcatDatasetWithIndex([d1, d2])
def __init__(self, size): d1 = CelebAValidation(size=size, keys=["image"]) d2 = CelebAHQValidation(size=size, keys=["image"]) d3 = FFHQValidation(size=size, keys=["image"]) self.data = ConcatDatasetWithIndex([d1, d2, d3])
def __init__(self, size): super().__init__() d1 = ConcatDataset([FFHQValidation(size=size, keys=["image"]), CelebAHQValidation(size=size, keys=["image"])]) d2 = AnimeFacesValidation(size=size, keys=["image"]) self.data = ConcatDatasetWithIndex([d1, d2])
def __init__(self, size): d1 = CelebAHQValidation(size=size) d2 = FFHQValidation(size=size) self.data = ConcatDatasetWithIndex([d1, d2])