图像预处理:颜色扭曲增强 transforms.ColorJitter(brightness=.5, hue=.3)

图像预处理:颜色扭曲增强 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'].head(200000), train_label['target'].head(200000),
		transforms.Compose([
			transforms.Resize((256, 256)),
			transforms.RandomHorizontalFlip(),
			transforms.RandomVerticalFlip(),
			transforms.ColorJitter(brightness=.5, hue=.3), #--颜色扭曲
			transforms.ToTensor(),
			transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
		])
	), batch_size=40, shuffle=True, num_workers=4, pin_memory=True
)

使用了 PyTorch 中的 torch.utils.data.DataLoader 来创建了一个训练数据加载器 train_loader。

在创建数据加载器时,首先指定了数据集为 FFDIDataset,并传入了训练数据的路径和标签。接着使用了一系列图像预处理的操作,比如调整尺寸、随机水平翻转、随机垂直翻转、颜色增强等,最后将图像转换为张量,并对图像进行了归一化处理。

此外,还设置了参数 batch_size=40 来指定每个批次的样本数量,shuffle=True 表示在每个 epoch 开始前打乱数据顺序,num_workers=4 表示使用 4 个子进程来加载数据以加快速度,pin_memory=True 表示将数据加载到 GPU 内存中以加速训练过程。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值