pytorch自定义dataloder的时候，返回参数

比如我想在返回矩阵同时，也返回地址。只需要将这两个信息用字典封装起来一起返回。
在train的时候用XXX[‘path’]和XXX[‘data’]调用即可。（for i,XXX in in enumerate(train_loader)）

class Tche_loader(data.Dataset):

	def __init__(self, lowlight_images_path):

		self.train_list = sorted(glob.glob(lowlight_images_path + "*.jpg"))

		self.data_list = self.train_list
		print("Total training examples:", len(self.train_list))

	def __getitem__(self, index):
 
		data_lowlight_path = self.data_list[index]

		data_lowlight = Image.open(data_lowlight_path)
		

		data_lowlight = (np.asarray(data_lowlight)/255.0) 
		data_lowlight = torch.from_numpy(data_lowlight).float()

		return {'path':data_lowlight_path,'data':data_lowlight.permute(2,0,1)}

	def __len__(self):
		return len(self.data_list)

any questions 请私信我