Pytorch学習ノート(三)-手書きデジタル認識
1.まず必要なパッケージをインポートします.ここでtorchvisionパッケージは主にデータの処理、インポート、プレビューを実現します.
import torch
from torchvision import datasets, transforms
from torch.autograd import Variable
2.torchvision datasets ， MNIST、COCO、ImageNet、CIFCAR， ：
# Download the datasets
data_train = datasets.MNIST(root = "./data/",
transform=transform,
train = True,
download = True)
data_test = datasets.MNIST(root = "./data/",
transform=transform,
train = False)
：root ，train ，transfrom ， ：
# Set the transform format
transform = transforms.Compose([transforms.ToTensor(),
])
3.transforms ：
transforms ， Tensor ， 。 ， ， 。（ ）
Compose ， ， 。 ：
torchvision.transforms.Resize(h,w)： ；
torchvision.transforms.Scale(h,w) ：
torchvision.transforms.CenterCrop(h,w) ： ，
torchvision.transforms.RondomHorizontalFlip(rate) ：
torchvision.transforms.RondomVerticalFlip(rate) ：
torchvision.transforms.ToTensor()
torchvision.transforms.ToPILImage()