Pytoch.Imagefloader经典数据集:hymenoptera数据集下载太慢解决方法、数据集介绍和使用

Pytoch经典数据集,蜜蜂和蚂蚁数据集(Hymenoptera)

参考资料Pytoch Hymenoptera

因网站资源托管在Amazon服务器上,所以下载速度可能奇慢无比,本人已将该数据集上传,下面提供国内下载:
hymenoptera_data数据集,[国内快速下载]

这个数据集包括两类图片, beesants, 这些数据都被处理成了可以使用ImageFolder 来读取的格式。我们只需要把data_dir设置成数据的根目录,然后把model_name设置成我们想要使用的训练模型,如:
resnet , alexnet, vgg16等。

Pytoch.Imagefloader的其他参数有

  • num_classes表示数据集分类的类别数
  • batch_size
  • num_epochs
  • feature_extract表示我们训练的时候使用fine tuning还是feature extraction方法。如果feature_extract = False,整个模型都会被同时更新。如果feature_extract = True,只有模型的最后一层被更新。

下载链接:https://download.csdn.net/download/weixin_46233323/12182815

发布了13 篇原创文章 · 获赞 24 · 访问量 1256

猜你喜欢

转载自blog.csdn.net/weixin_46233323/article/details/104587813