目标效果:
代码:
import numpy as np
import torch
from torchvision import datasets
import torchvision.transforms as transforms
from torch.utils.data.sampler import SubsetRandomSampler
import matplotlib.pyplot as plt
#加载数据
transform=transforms.Compose([
transforms.ToTensor(), #数据集加载时,默认的图片格式是 numpy,所以通过 transforms 转换成 Tensor,图像范围[0, 255] -> [0.0,1.0]
transforms.Normalize((0.5,0.5,0.5),(0.5,0.5,0.5)) #使用公式进行归一化channel=