如何在张量流中进行图像旋转？答案

【问题标题】：How to do image rotation in tensorflow?如何在张量流中进行图像旋转？
【发布时间】：2018-03-19 19:45:46
【问题描述】：

我想在我自己的数据集上做一些严肃的图像增强，因为它真的很小。 TensorFlow 提供了图像库来做所有的增强，非常方便。但是没有“旋转”功能。我知道here 跟踪了一个问题，但它很长而且看起来很复杂。有没有简单的方法/功能在张量流中进行图像旋转？非常感谢您。

【问题讨论】：

【解决方案1】：

如果您想使用自己的图像旋转，您可以使用 py_func、PIL 和您自己的队列处理器来实现。或者，您可以编写自己的队列来提供示例。另一种丑陋的方法是在训练进行时自动（写入+重命名）旋转这些图像并使用相同的文件名将它们写入磁盘。

无论您做什么，请记住我们是分批训练的，因此请确保您一次提供一个轮换，而不是同一图像的所有变体都连续提供（shuffle(rotations([image]) 不是rotations(shuffle([image]))），否则训练会出现异常。

【讨论】：

【解决方案2】：

我一直在寻找同样的东西 - 现在这在 Tensorflow 中实现为：

tf.contrib.image.rotate(images,
                        angles,
                        interpolation='NEAREST')

在此处查看doc (r1.4)。

【讨论】：