pytorch - 如何从 pytorch 中的列表构建数据加载器

Question

我得到了一个图像列表和另一个标签列表。图片：shape(64, 64, 3) 标签：shape(64,)。实际上，标签是 64 维的连续值。

import pickle

file_name = "trainData.pkl"
open_file = open(file_name, "rb")
loaded_list = pickle.load(open_file)
open_file.close()

print(loaded_list[0])
print(loaded_list[0].shape)

file_name = "trainLabel.pkl"
open_file = open(file_name, "rb")
loaded_list = pickle.load(open_file)
open_file.close()

print(loaded_list[0])
print(loaded_list[0].shape)

标签： pytorchresnetpytorch-dataloader