如何在Keras中使用ImageDataGenerator调整标签？

Question

我想使用Keras的ImageDataGenerator增强我的数据集，以便与model.fit_generator()一起使用。 我发现我可以随机翻转图像。 对于翻转的图像，我需要修改相应的标签。 我该如何做呢？

编辑：我正在进行回归，而不是分类，因此如果图像被翻转，我需要调整标签。 实际图像来自自动驾驶汽车模拟器，标签是转向角度。 如果我水平翻转图像，我需要取反转向角度。

- royco

---

你有兴趣尝试黑客解决方案吗？ - Marcin Możejko

当然可以！ - royco

你的dim_ordering是什么？ - Marcin Możejko

dim_ordering='tf' - royco

Answer 1

您可以尝试以下操作：

import numpy

def fliping_gen(image_generator, flip_p=0.5):
    for x, y in image_generator:
        flip_selector = numpy.random.binomial(1, flip_p, size=x.shape[0]) == 1
        x[flip_selector,:,:,:] = x[flip_selector,:,::-1,:]
        y[flip_selector] = (-1) * y[flip_selector]
        yield x, y

- Marcin Możejko