计算图片的均值和方差用图片的归一化取值

计算图片的均值和方差用图片的归一化取值

注意：使用这种方法的前提是进行了数据批量化操作，需要使用神经网络库，torch，DataLoader

def getStat(data):
    print(len(data))
    loader = torch.utils.data.DataLoader(data, batch_size=1, shuffle=False,num_workers=0,pin_memory=True)
    mean = torch.zeros(3)
    std = torch.zeros(3)
    for X, _ in loader:
        for d in range(3):
            mean[d] += X[:, d, :, :].mean()
            std[d] += X[:, d, :, :].std()
    mean.div_(len(data))
    std.div_(len(data))
    return list(mean.numpy()), list(std.numpy())
 
mean,std = getstat(train_data)#调用getstat
mean_,std_ = getstat(test_data)
print(mean,std)
print(mean_,std_)

此方法可以计算3通道和一个通道的均值和方差。