如何解决如何使用文件数据集的相同顺序创建文件夹中文件名的字典ImageFolder创建数据集
我有一个包含测试3个文件[img1.jpg,img2.jpg,img3.jpg]的文件夹,用于预测。我使用ImageFolder和Dataloader类来服务我的预训练Pytorch模型。
我假设ImageFolder类以升序格式创建数据集。
以下代码为我提供了以results ={0:'class0',1:'class1'..}.
for idx,(data,labels) in enumerate(test_loader):
output = model(data)
pred = torch.argmax(output,1)
pred = pred.detach().cpu()
if pred.item() == 0:
results[idx] = 'class0'
else:
results[idx] = 'class1'
由于我需要为{'image1.jpg': 'class1',...}
提供JSON文件,如何将文件的数据集名称提供给字典?
os.listdir(path)由于运动方式不正确-返回文件的随机顺序,但是创建数据集时我对顺序很敏感。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。