首先需要会一点点Python字典的知识,可以参考菜鸟教程注意一下人家字典的元素调用方式。
然后对着代码分析就可以了。
上代码
# 这里就运用了字典结构。分为两部分train和val,各自的内容用“:”加到后面。然后两
#部分之间会用一个小逗号“,”隔开。当然冒号后面代表的是分别给train和val数据集的操作
#就比如给train先后进行随机裁剪,转换成Tensor向量,然后进行规范化操作,这三个操作
#用Compose连起来,代表会依次执行。
data_transforms = {
'train': transforms.Compose([
transforms.RandomResizedCrop(input_size),
transforms.RandomHorizontalFlip(),
transforms.To