图像预处理:颜色扭曲增强 transforms.ColorJitter(brightness=.5, hue=.3)
transforms.ColorJitter的使用和作用
transforms.ColorJitter 是 PyTorch 中用于进行颜色扭曲(Color Jittering)的一个预处理操作。在这里,你使用了 transforms.ColorJitter(brightness=.5, hue=.3) 对图像进行了颜色扭曲处理。
具体来说,brightness=.5 表示调整亮度的范围为 [-0.5, 0.5],而 hue=.3 表示调整色调的范围为 [-0.3, 0.3]。这样的操作可以增加数据的多样性,使模型更加鲁棒,并且可以帮助模型学习不同光照条件下的图像特征。
train_loader = torch.utils.data.DataLoader(
FFDIDataset(train_label['path'