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)[source]¶
隨機改變圖片或影片的亮度、對比度、飽和度和色相。
如果輸入是
torch.Tensor，預期其形狀為 […, 1 或 3, H, W]，其中 … 表示任意數量的 leading dimensions。如果 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的範例- 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]][source]¶
取得隨機轉換要應用於影像的參數。
- 參數:
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 以關閉轉換。
- Returns:
用於套用隨機轉換的參數及其隨機順序。
- Return type:
