深度学习 蜜蜂数据集下载(Hymenoptera)
这个数据集包括两类图片, bees 和 ants, 这些数据都被处理成了可以使用ImageFolder 来读取的格式。我们只需要把data_dir设置成数据的根目录,然后把model_name设置成我们想要使用的训练模型下载链接:https://download.csdn.net/download/weixin_46233323/12182815
·
Pytoch经典数据集,蜜蜂和蚂蚁数据集(Hymenoptera)
参考资料:Pytoch Hymenoptera
因网站资源托管在Amazon服务器上,所以下载速度可能奇慢无比,本人已将该数据集上传,下面提供国内下载:
hymenoptera_data数据集,[国内快速下载]
这个数据集包括两类图片, bees 和 ants, 这些数据都被处理成了可以使用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
更多推荐
已为社区贡献1条内容
所有评论(0)