Cifar10数据集可视化(pytorch)

1.显示效果

上图显示为1batch的数据集可视化结果，batch_size可以自定义。这里设定line21 shuffle=True,所以每次运行结果都不相同。

2.配置config

matplotlib        3.3.2
numpy             1.19.2
torch             1.6.0+cpu
torchvision       0.7.0+cpu

3.代码

import numpy as np

import torch

import torchvision
import torchvision.transforms as transforms

import matplotlib.pyplot as plt
#加载数据
#train_set:50000张图片
classes = ['plane', 'car'</