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()