【笔记】加载数据过程中在batch_size中两种图片读取方式:1. next(iter(data_loader))[0] 2. __getitem__ 中顺序读取图片 两者读取图片的顺序是一致的

准确的说,是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版权协议,转载请附上原文出处链接和本声明。