1. 基本功能函数使用讲解
import torchvision
# root 数据集位置,train 训练集或测试集,download 是否下载
train_set = torchvision.datasets.CIFAR10( root ="./dataset",train=True,download=True)
test_set = torchvision.datasets.CIFAR10(root="./dataset",train=False,download=True)
print(test_set[0])
# (<PIL.Image.Image image mode=RGB size=32x32 at 0x7F52D13E00D0>, 3)
print(test_set.classes)
# ['airplane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck']
img, target = test_set[0]
print(test_set.classes[target])
# cat
img.show()
# 显示img
2. dataset + transform 运用
transform 把图片转为tensor ,并用tensorboard 显示
from torchvision import transforms
fr