使用python和tensorflow旋转90°、180°和270°

Question

我想将图像随机旋转 90、180 和 270 度，即 90 度的倍数。

目前，我正在使用ImageDatagenerator来增强我的数据：

train_dataGen = ImageDataGenerator(rescale=None,horizontal_flip=True,rotation_range=90,
                                   vertical_flip=True)

如果rotation_rangevariable等于90，就像上面的代码一样，它是否只旋转90度？或者它执行 90 倍数的旋转？

Answer 1

def rotate_image(image):
    return np.rot90(image, np.random.choice([-1, 0, 1]))

train_dataGen = ImageDataGenerator(
    preprocessing_function=rotate_image)

此函数将旋转 -90 度、0 度或 90 度。