python - 使用 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_range 变量等于 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 度。