pytorch（一）pytorch&torchvision介绍与安装

torchvision介绍

torchvision 是PyTorch中专门用来处理图像的库。这个包中有四个大类。

torchvision.datasets
torchvision.models
torchvision.transforms
torchvision.utils

torchvision.datasets

torchvision.datasets 是用来进行数据加载的，PyTorch团队在这个包中帮我们提前处理好了一些图像的公开数据集。

MNISTCOCO
Captions
Detection
LSUN
ImageFolder
Imagenet-12
CIFAR
STL10
SVHN
PhotoTour
···

下载数据集方法

train_data = torchvision.datasets.MNIST(
    root="./mnist",  # 设置数据集的根目录
    train=True,  # 是否是训练集
    transform=trans,  # 对数据进行转换
    download=DOWNLOAD_MNIST
)
# 第二个参数是数据分块之后每一个块的大小，第三个参数是是否大乱数据
train_loader = DataLoader(train_data, batch_size=BATCH_SIZE, shuffle=True)

test_data = torchvision.datasets.MNIST(
    root=