如何获取在其上训练了pytorch模型的imagenet数据集

时间:2020-06-07 16:03:07

标签: pytorch torchvision imagenet

谁能告诉我如何下载训练pytorch火炬视觉模型并报告其Top-1错误的完整imagenet数据集?

我已经从Imagenet网站下载了Tiny-Imagenet,并使用了预训练的resnet-101模型,该模型仅提供18%的Top-1准确性。

1 个答案:

答案 0 :(得分:1)

http://www.image-net.org/下载ImageNet数据集(您必须登录)

然后,您应该将验证图像移动到带标签的子文件夹,这可以使用以下外壳程序脚本自动完成: https://raw.githubusercontent.com/soumith/imagenetloader.torch/master/valprep.sh