在批处理上应用图像增强

时间:2016-06-28 22:29:47

标签: tensorflow

如何将tf.image.random_flip_left_right等图像增强功能应用于一批图像?

这些函数要求输入大小为3级([width, height, channels]),而批量数据的级别为4([batch_size, width, height, channels]

1 个答案:

答案 0 :(得分:1)

您可以使用tf.train.batch构建批量图像。

看看CIFAR10示例如何处理其输入: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/models/image/cifar10/cifar10_input.py

您可以在此处找到有关CIFAR10示例的教程: https://www.tensorflow.org/versions/r0.9/tutorials/deep_cnn/index.html

希望有所帮助!