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

【问题标题】：How to adjust labels with ImageDataGenerator in Keras?如何在 Keras 中使用 ImageDataGenerator 调整标签？
【发布时间】：2017-06-18 16:00:27
【问题描述】：

我想用 Keras 的 ImageDataGenerator 扩充我的数据集，以便与 model.fit_generator() 一起使用。我看到我可以随机翻转图像。对于翻转的图像，我需要修改相应的标签。我该怎么做？

编辑：我在做回归，而不是分类，所以如果图像被翻转，我需要调整标签。实际图像来自自动驾驶汽车模拟器，标签是转向角。如果我水平翻转图像，我需要否定转向角。

【问题讨论】：

对黑客解决方案感兴趣？
@MarcinMożejko 当然！
你的 dim_ordering 是什么？
dim_ordering='tf'

标签： python machine-learning neural-network deep-learning keras

【解决方案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

【讨论】：