Exemple #1
0
 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)