将数据集与transforms结合

  • 可以直接在PyCharm中下载数据集
  1. import torchvision
  2. from torch.utils.tensorboard import SummaryWriter
  3. dataset_transform = torchvision.transforms.Compose([
  4. torchvision.transforms.ToTensor()
  5. ])
  6. # train_dataset = torchvision.datasets.CIFAR10(root="./download_data",train=True,download=True)
  7. # test_dataset = torchvision.datasets.CIFAR10(root="./download_data",train=False,download=True)
  8. # 下载数据集,这其实是一个下载接口
  9. train_dataset = torchvision.datasets.CIFAR10(root="./download_data",train=True,transform=dataset_transform,download=True)
  10. test_dataset = torchvision.datasets.CIFAR10(root="./download_data",train=False,transform=dataset_transform,download=True)
  11. '''
  12. print(test_dataset[0])
  13. print(test_dataset.classes)
  14. img,target = test_dataset[0]
  15. print(img)
  16. print(target)
  17. print(test_dataset.classes[target])
  18. img.show()
  19. '''
  20. writer = SummaryWriter("test_data")
  21. for i in range(10):
  22. img,target = test_dataset[i]
  23. writer.add_image("dataset",img,i)
  24. writer.close()