DataLoader无法返回文件名

想把网络跑出来的图像放入matlab中处理（好吧因为不知该怎么在python中将bgr->rgb），但是在matlab中批量处理，文件名我不想改变。

所以可以从下面代码来操作：

name_list =[]
for filename in os.listdir(path=r'test_image/ori'):
    name_list.append(filename)

for test_image in enumerate(test_dataloader):
    idx = test_image[0]     # 这里返回的是第idx张图片
    ...

    output = net(input)
    output_name = name_list[idx]
    save_image(output, output_name)

这样就能保持文件名不变的保存生成的图像了！