为什么我在同一个索引处得到不同的图像？

Question

我有以下代码部分：

images = []
image_labels = []

for i, data in enumerate(train_loader,0):
        inputs, labels = data
        inputs, labels = inputs.to(device), labels.to(device)
        inputs, labels = inputs.float(), labels.float()
        images.append(inputs)
        image_labels.append(labels)

image = images[7]
image = image[0,...].permute([1,2,0])
image = image.numpy()
image = (image * 255).astype(np.uint8)
img = Image.fromarray(image,'RGB')
img.show()

如您所见，我试图在索引 7 处显示图像。但是，每次运行代码时，虽然使用相同的索引，但都会显示不同的图像，这是为什么呢？

显示的图像也像降级了，质量低于原始图像。

对此有什么想法吗？

谢谢。

Answer 1

我最好的选择是您将DataLoader 的shuffle 选项设置为True，在这种情况下，它会导致不同的图像出现在索引7。每次遍历迭代器时，用于访问底层数据集的索引序列都会有所不同。