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 或 3, H, W] 的形狀，其中 ... 表示任意數量的前導維度。如果 img 是 PIL Image，則不支援模式 “1”、“I”、“F” 以及帶透明度（Alpha 通道）的模式。

引數:

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

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

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

• hue (float 或 tuple of python:float (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 影像或張量

static get_params(brightness: Optional[list[float]], contrast: Optional[list[float]], saturation: Optional[list[float]], hue: Optional[list[float]]) → tuple[torch.Tensor, Optional[float], Optional[float], Optional[float], Optional[float]]

獲取應用於影像的隨機變換的引數。

引數:

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

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

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

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

返回:

用於應用隨機變換及其隨機順序的引數。

返回型別:

元組