PyTorch - transforms.ColorJitter 改变图像的属性：亮度（brightness）、对比度（contrast）、饱和度（saturation）和色调（hue)

PyTorch - transforms.ColorJitter 改变图像的属性：亮度（brightness）、对比度（contrast）、饱和度（saturation）和色调（hue)

flyfish

jitter的意思

v. 紧张不安 / 抖动 / 战战兢兢 / 神经过敏
n. 紧张不安 / 晃动 / 偏移 / 振动
这里抖动或者偏移 是随机变化的含义

下面以亮度brightness举例，假设brightness设置为0.5
brightness_change = transforms.ColorJitter(brightness=0.5)
它的含义是将图像的亮度随机变化为原图亮度的50%（1−0.5）∼150%（1+0.5）
说明中的[max(0, 1 - brightness), 1 + brightness]就是 [0.5 , 1.5]
其他参数可以举一反三

brightness（float或 float类型元组(min, max)）– 亮度的偏移幅度。
brightness_factor从[max(0, 1 - brightness), 1 + brightness]中随机采样产生。应当是非负数。

contrast（float或 float类型元组(min, max)）– 对比度偏移幅度。
contrast_factor从[max(0, 1 - contrast), 1 + contrast]中随机采样产生。应当是非负数。

saturation（float或 float类型元组(min, max)）– 饱和度偏移幅度。
saturation_factor从[max(0, 1 - saturation), 1 + saturation]中随机采样产生。应当是非负数。

hue（float或 float类型元组(min, max)）– 色相偏移幅度。
hue_factor从[-hue, hue]中随机采样产生，其值应当满足0<= hue <= 0.5或-0.5 <= min <= max <= 0.5

代码示例

import numpy as np
import cv2
import os
import torch
import math
import torchvision.transforms as transforms
from PIL import Image


# 单独设置
# 随机改变图像的亮度
brightness_change = transforms.ColorJitter(brightness=0.5)
# 随机改变图像的色调
hue_change = transforms.ColorJitter(hue=0.5)
# 随机改变图像的对比度
contrast_change = transforms.ColorJitter(contrast=0.5)

# 综合设置
color_aug = transforms.ColorJitter(brightness=0.5, contrast=0.5, saturation=0.5, hue=0.5)

transform = transforms.Compose([
        brightness_change,
        hue_change,
        contrast_change,
    ])