如何在张量流中进行图像旋转?

时间:2016-11-12 06:12:15

标签: tensorflow image-rotation

我想在我自己的数据集上做一些严肃的图像增强,因为它非常小。 Tensorflow提供了图像库来完成所有扩充,非常方便。但是没有"旋转"功能。我知道有一个跟踪here的问题,但它太长了,看起来很复杂。在张量流中有没有简单的方法/功能来进行图像旋转?非常感谢你。

2 个答案:

答案 0 :(得分:2)

如果您想使用自己的图像旋转,可以使用py_func,PIL和您自己的队列处理器来完成。或者,您可以编写自己的队列来提供示例。另一种丑陋的方式是,原子地(写入+重命名)旋转这些图像并将它们写在具有相同文件名的磁盘上,同时进行训练。

无论你做什么,请记住我们分批训练,所以请确保你一次一个地轮换,而不是同一个图像的所有变体连续(shuffle(rotations([image])不是rotations(shuffle([image]))),否则训练会行为不端。

答案 1 :(得分:1)

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

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

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