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

更多推荐