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 或 python:float 的元組 (min, max)) – 亮度抖動的程度。brightness_factor 從 [max(0, 1 - brightness), 1 + brightness] 或給定的 [min, max] 中均勻選取。應為非負數。

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

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

• hue (float 或 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 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 的元組 (min, max), 可選) – brightness_factor 均勻選取的範圍。傳入 None 以關閉此變換。

• contrast (python:float 的元組 (min, max), 可選) – contrast_factor 均勻選取的範圍。傳入 None 以關閉此變換。

• saturation (python:float 的元組 (min, max), 可選) – saturation_factor 均勻選取的範圍。傳入 None 以關閉此變換。

• hue (python:float 的元組 (min, max), 可選) – hue_factor 均勻選取的範圍。傳入 None 以關閉此變換。

返回：

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

返回型別：

tuple