我应将CSV测试文件放在PyTorch数据加载器中的什么位置？

Question

假设我有test.csv

   filename
1  a.jpg

2  b.jpb

然后我有测试图像文件夹
/ test

test_dataset = torchvision.datasets.ImageFolder(root= path + 'test/',transform=trans)

这将带来所有测试文件

如果我想在完成培训后制作提交文件，该如何链接测试文件夹的名称和submitting.csv文件名？

%%time
from torch.autograd import Variable
results = []
with torch.no_grad():
    model.eval()
    print('start')
    for num, data in enumerate(test_loader):
        #print(num)
        imgs, label = data
        imgs,labels = imgs.to(device), label.to(device)
        test = Variable(imgs)
        output = model(test)
        ps = torch.exp(output)
        top_p, top_class = ps.topk(1, dim = 1)
        #print(top_class)
        results += top_class.cpu().numpy().tolist()


predictions = np.array(results).flatten()

array（[0，0，0，0，0，0，0，0，0，0，0，0，0，1，1，0，0，0，0，0，0， 0， 1，0，0，0，1，0，0，0，0，1，0，0，1，0，1，0，0，0，0，0，0，0， 0，0，0，0，0，0，1，0，0，0，0，1，0，0，0，0，0，0，1，0，0，0， 0，0，0，0，0，1，0，0，1，1，0，0，0，0，0，0，0，0，0，0，0，0， 0，0，0，0，0，0，0，0，0，0，0，0，0，0，1，0，0，0，0，0，0，0， 1，0，0，1，0，0，0，0，0，1，1，0，0，0，0，0，0，0，0，1，0，0， 0，0，0，0，0，0，0，0，1，0，0，0，1，0，0，0，0，0，0，0，0，0， 0，0，0，0，1，0，0，0，0，0，0，0，0，1，0，0，0，0，0，0，0，0， 0，0，0，0，0，0，0，1，1，0，0，1，0，0，0，0，0，1，0，0，0，0， 0、1、0、0、0、0、0、0、0、0、0、0、0、0、0、1、0、0、0、0、0， 0，0，0，0，0，0，0，1，0，0，1，1，0，0，0，1，0，0，0，0，0，0， 1，0，0，0，1，1，1，0，0，0，0，0，0，0，1，1，0，0，0，0，0，0， 0，0，0，0，1，0，0，0，0，0，0，0，0，0，0，0，0，0，0，0，0，0]）

我怎么知道哪个结果来自哪个文件？