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