zhaoyangtcc的博客

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

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