torch中内置的数据集:

ps:在修改代码时,之前研究过一些数据集可以通过代码在训练的时候进行下载,刚刚得知这些数据集是torch中内置的数据集,想自己总结记录一下

Torchvision 中的数据集:

MNIST:

手写图像组成的数据集,60,000 张训练图像和 10,000 张测试图像(也是联邦学习实验中经常使用的数据集)在训练的时候很容易达到很高的精度

train_dataset = datasets.MNIST(data_dir, train=True, download=True,
                                      transform=apply_transform)
test_dataset = datasets.MNIST(data_dir, train=False, download=True,
                                      transform=apply_transform)

Fashion MNIST:

类似于mnist数据集,是一组28x28灰度的衣服图像,训练和测试样本数分别为60,000和10,000

CIFAR:

CIFAR10和CIFAR100。CIFAR10 由 10 个不同标签的图像组成,而 CIFAR100 有 100 个不同的类。这些包括常见的图像,如卡车、青蛙、船、汽车、鹿等。

train_dataset = datasets.CIFAR10(data_dir, train=True, download=True,
                                       transform=apply_transform)

test_dataset = datasets.CIFAR10(data_dir, train=False, download=True,
                                      transform=apply_transform)

COCO:

COCO数据集包含超过 100,000 个日常对象,如人、瓶子、文具、书籍等

EMNIST:

包括数字和字母的图像组成。适合处理从图像中识别文本

IMAGE-NET:

分布在 10,000 个类别中的超过 120 万张图像组成

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值