我是Pytorch的新手,并不是CNN的专家.我已经用他们提供Tutorial Pytorch的教程做了一个成功的分类器,但我真的不明白我在加载数据时正在做什么.
他们为训练做了一些数据增加和规范化,但是当我尝试修改参数时,代码不起作用.
# Data augmentation and normalization for training
# Just normalization for validation
data_transforms = {
'train': transforms.Compose([
transforms.RandomResizedCrop(224),
transforms.RandomHorizontalFlip(),
transforms.ToTensor(),
transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
]),
'val': transforms.Compose([
transforms.Resize(256),
transforms.CenterCrop(224),
transforms.ToTensor(),
transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
]),
}
Run Code Online (Sandbox Code Playgroud)
我是否在扩展我的训练数据集?我没有看到数据扩充.
为什么我修改transforms.RandomResizedCrop(224)的值,数据加载停止工作?
我是否还需要转换测试数据集?
我对他们所做的数据转换有点困惑.