Как настроить ярлыки с ImageDataGenerator в Keras?

Question

Я хочу увеличить свой набор данных с помощью Keras's ImageDataGenerator для использования с model.fit_generator(). Я вижу, что я могу случайным образом переворачивать изображения. Для перевернутых изображений мне нужно изменить соответствующую метку. Как я могу это сделать?

EDIT: Я делаю регрессию, а не классификацию, поэтому, если изображение перевернуто, мне нужно отрегулировать ярлык. Фактические изображения взяты из самозапускающегося симулятора автомобиля, а наклейки - углы поворота. Если я горизонтально перевернул изображение, мне нужно отменить угол поворота.

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