Python cifar_image_augmentation Exemples

Exemple #1

Fichier : image.py Projet : zeyu-h/tensor2tensor

 def preprocess_example(self, example, mode, unused_hparams):
   example["inputs"].set_shape([_CIFAR10_IMAGE_SIZE, _CIFAR10_IMAGE_SIZE, 3])
   if mode == tf.estimator.ModeKeys.TRAIN:
     example["inputs"] = common_layers.cifar_image_augmentation(
         example["inputs"])
   example["inputs"] = tf.to_int64(example["inputs"])
   return example

Exemple #2

 def preprocess_example(self, example, mode, unused_hparams):
     example["inputs"].set_shape(
         [_CIFAR10_IMAGE_SIZE, _CIFAR10_IMAGE_SIZE, 3])
     if mode == tf.estimator.ModeKeys.TRAIN:
         example["inputs"] = common_layers.cifar_image_augmentation(
             example["inputs"])
     example["inputs"] = tf.to_int64(example["inputs"])
     return example

Exemple #3

 def preprocess_example(self, example, mode, unused_hparams):
     if mode == tf.estimator.ModeKeys.TRAIN:
         example["inputs"] = common_layers.cifar_image_augmentation(
             example["inputs"])
     return example

Exemple #4

Fichier : image.py Projet : ins0m/tensor2tensor

 def preprocess_examples(self, examples, mode):
     if mode == tf.contrib.learn.ModeKeys.TRAIN:
         examples["inputs"] = common_layers.cifar_image_augmentation(
             examples["inputs"])
     return examples