ColorJitter¶
- class torchvision.transforms.v2.ColorJitter(brightness: Optional[Union[float, Sequence[float]]] = None, contrast: Optional[Union[float, Sequence[float]]] = None, saturation: Optional[Union[float, Sequence[float]]] = None, hue: Optional[Union[float, Sequence[float]]] = None)[源]¶
隨機改變影像或影片的亮度、對比度、飽和度和色調。
如果輸入是
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的示例- 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 以關閉變換。
- 返回:
用於應用隨機變換的引數及其隨機順序。
- 返回型別:
