torchvision.datasets 的使用(附代码)

一.torchvision.datasets是什么?作用?

很明显,datasets是torchvision工具集中的一个工具

初步理解其是调用官方数据集的一种方式,其中存在很多已经开源的数据集,供我们学习使用.

二.使用方法

已知CIFAR10是datasets可以调用的关于图像的数据集.我们可以尝试使用,并利用tensorboard进行可视化

• 第一步:

  引用torchvision模块,和tensorboardX

  import torchvision
  from tensorboardX import SummaryWriter
  

• 第二步:

  获取数据集的数据

  test_data = torchvision.datasets.CIFAR10(
      root='../BigData',
      transform=torchvision.transforms.ToTensor(),
      train=False,
      download=True
  )
  

• 第三步:

  在tensorboard中展示,由于该数据集的图像像素为32*32,所以比较模糊.

  writer = SummaryWriter('LOGS/008log')
  for i in range(10):
      img, target = test_data[i]
      writer.add_image('test_set', img, i)
  
  writer.close()