如何调整ImageDataGenerator的旋转角度范围？能否设置θ_min至θ_max区间旋转？

当然可以！默认的rotation_range参数确实只能让图像在0到指定角度范围内随机旋转，但如果你想要限定在50到60度这样的特定区间，我们可以通过自定义预处理函数轻松实现——完全支持在任意θ_min与θ_max之间进行随机旋转。

实现步骤 #

核心思路是利用ImageDataGenerator的preprocessing_function参数，传入一个自定义的旋转函数，在函数内生成指定区间的随机角度并应用到图像上。

1. 导入必要模块

import numpy as np
from tensorflow.keras.preprocessing.image import ImageDataGenerator
from tensorflow.keras.preprocessing.image import apply_affine_transform

2. 定义自定义旋转函数
   这个函数会随机生成50-60度之间的角度，然后对输入图像执行旋转操作：

def custom_range_rotation(image):
    # 在50°到60°之间随机生成旋转角度
    random_angle = np.random.uniform(low=50, high=60)
    # 应用旋转变换，fill_mode可根据需求调整（如'nearest'/'constant'/'reflect'等）
    rotated_image = apply_affine_transform(
        image,
        theta=random_angle,
        fill_mode='nearest',
        cval=0  # 如果fill_mode选'constant'，这里设置填充的常数值
    )
    return rotated_image

3. 创建带自定义旋转的ImageDataGenerator
   将自定义函数传入preprocessing_function参数即可，还能同时结合其他数据增强参数：

datagen = ImageDataGenerator(
    preprocessing_function=custom_range_rotation,
    width_shift_range=0.1,
    height_shift_range=0.1
)

额外说明 #

• 这种方法灵活性拉满，你可以把low和high的值改成任何需要的区间（比如-30到30度实现左右对称旋转）。
• apply_affine_transform支持处理单通道（灰度图）和多通道（RGB）图像，无需额外修改。
• 如果需要更复杂的逻辑（比如根据图像类别调整旋转区间），也可以直接在自定义函数里扩展。

内容的提问来源于stack exchange，提问作者Didier