zhaoyangtcc的博客

原创 PyTorch深度学习多分类—刘二大人

import torch from torchvision import datasets from torchvision import transforms from torch.utils.data import DataLoader # 数据准备 batch_size = 64 # Compose组合多个transform操作，Normalize将数据进行标准版，1是均值，2是方差 # 图像PIL或者opencv都是（H, W, C），利用ToTensor转换为（C, H, W） transfo.