def prepare_training_generators(self): self._train_generator = COCOGenerator( self.train_image_dir, self.train_annotation_path, batch_size=self.batch_size, shuffle=True, label_encoder=self.input_encoder, dct=True, split_cbcr=True, transforms=self.train_transformations) self._validation_generator = COCOGenerator( self.validation_image_dir, self.validation_annotation_path, batch_size=self.batch_size, shuffle=True, label_encoder=self.input_encoder, dct=True, split_cbcr=True, transforms=self.validation_transformations) self.validation_steps = len(self._validation_generator)