ColorJitter

class torchvision.transforms.ColorJitter(brightness: Union[float, Tuple[float, float]] = 0, contrast: Union[float, Tuple[float, float]] = 0, saturation: Union[float, Tuple[float, float]] = 0, hue: Union[float, Tuple[float, float]] = 0)

隨機改變影像的亮度、對比度、飽和度和色相。如果影像為 torch Tensor，則預期其形狀為 […, 1 or 3, H, W]，其中 … 表示任意數量的領先維度。如果 img 是 PIL Image，則不支援模式 “1”、“I”、“F” 和具有透明度（alpha 通道）的模式。

參數:

• brightness (float 或 python:float 的 tuple (min, max)) – 亮度抖動的程度。brightness_factor 從 [max(0, 1 - brightness), 1 + brightness] 或給定的 [min, max] 中均勻選擇。應該是非負數。

• contrast (float 或 python:float 的 tuple (min, max)) – 對比度抖動的程度。contrast_factor 從 [max(0, 1 - contrast), 1 + contrast] 或給定的 [min, max] 中均勻選擇。應該是非負數。

• saturation (float 或 python:float 的 tuple (min, max)) – 飽和度抖動的程度。saturation_factor 從 [max(0, 1 - saturation), 1 + saturation] 或給定的 [min, max] 中均勻選擇。應該是非負數。

• hue (float 或 python:float 的 tuple (min, max)) – 色相抖動的程度。hue_factor 從 [-hue, hue] 或給定的 [min, max] 中均勻選擇。應具有 0<= hue <= 0.5 或 -0.5 <= min <= max <= 0.5。為了抖動色相，輸入影像的像素值必須是非負數，才能轉換為 HSV 空間；因此，如果您將影像歸一化為具有負值的區間，或者在使用此函數之前使用產生負值的插值，則它將不起作用。

使用 ColorJitter 的範例

轉換的圖示

轉換的圖示

forward(img)

參數:

img (PIL Image 或 Tensor) – 輸入影像。

返回:

色彩抖動後的影像。

返回類型:

PIL Image 或 Tensor

static get_params(brightness: Optional[List[float]], contrast: Optional[List[float]], saturation: Optional[List[float]], hue: Optional[List[float]]) → Tuple[Tensor, Optional[float], Optional[float], Optional[float], Optional[float]]

取得要應用於影像的隨機轉換的參數。

參數:

• brightness (python:float 的 tuple (min, max), optional) – 從中均勻選擇 brightness_factor 的範圍。 傳遞 None 以關閉轉換。

• contrast (python:float 的 tuple (min, max), optional) – 從中均勻選擇 contrast_factor 的範圍。 傳遞 None 以關閉轉換。

• saturation (python:float 的 tuple (min, max), optional) – 從中均勻選擇 saturation_factor 的範圍。 傳遞 None 以關閉轉換。

• hue (python:float 的 tuple (min, max), optional) – 從中均勻選擇 hue_factor 的範圍。 傳遞 None 以關閉轉換。

返回:

用於應用隨機轉換的參數及其隨機順序。

返回類型:

tuple