def get_generator(self, batch_size=64, fullinfo=False):
if self.gen is None:
self.gen = DataGenerator(self.data, self.target_shape, with_augmentation=self.augment,
custom_augmentation=self.custom_augmentation, batch_size=batch_size,
num_classes=self.get_num_classes(), preprocessing=self.preprocessing,
fullinfo=fullinfo)
return self.gen
def get_generator(self, batch_size, doublelabel=False):
return DataGenerator(self.data,
target_shape=self.target_shape,
with_augmentation=self.augment,
custom_augmentation=self.custom_augmentation,
batch_size=batch_size,
num_classes=CLASSES,
preprocessing=self.preprocessing,
rebalance=self.rebalance,
rebalance_factor=self.rebalance_factor,
doublelabel=doublelabel)