准确的说,是next(iter(data_loader))[0]调用类MyDataSet中的__getitem__,所以两者的顺序是一致的。
eg1:
batch_size = 总图片数
方式一:
注意data的size()为 torch.Size([1200, 3, 100, 100])
方式二:
注意img的size()报错,'int' object is not callable;img的shape()报错,'tuple' object is not callable;img的shape为(100, 100, 3)
读取结果:
eg2:
batch_size= 1/10 的总图片数
方式一:
方式二:
读取结果:
版权声明:本文为nyist_yangguang原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。