深度学习 蜜蜂数据集下载(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

你可能感兴趣的:(Pytorch深度学习,深度学习,计算机视觉,大数据,pytorch,神经网络)