def prepare_training_generators(self):
self._train_generator = DCTGeneratorJPEG2DCT(
self.train_directory,
self.index_file,
batch_size=self.batch_size,
shuffle=True,
transforms=self.train_transformations)
self._validation_generator = DCTGeneratorJPEG2DCT(
self.validation_directory,
self.index_file,
batch_size=self.batch_size,
shuffle=True,
transforms=self.validation_transformations)
def prepare_testing_generator(self):
self._test_generator = DCTGeneratorJPEG2DCT(
self.test_directory,
self.index_file,
None,
1,
transforms=self.test_transformations)
def prepare_testing_generator(self):
self._test_generator = DCTGeneratorJPEG2DCT(
self.test_directory,
self.index_file,
None,
1,
shuffle=False,
transforms=self.test_transformations,
only_y=True)