软件开发相关文档,蚌埠seo招聘,网站建设服务商的网站建设流程,“网站建设：上海珍岛”计算图片的均值和方差用图片的归一化取值
注意#xff1a;使用这种方法的前提是进行了数据批量化操作#xff0c;需要使用神经网络库#xff0c;torch#xff0c;DataLoader
def getStat(data):print(len(data))loader torch.utils.data.DataLoader(data, batch_size1, …计算图片的均值和方差用图片的归一化取值
注意使用这种方法的前提是进行了数据批量化操作需要使用神经网络库torchDataLoader
def getStat(data):print(len(data))loader torch.utils.data.DataLoader(data, batch_size1, shuffleFalse,num_workers0,pin_memoryTrue)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通道和一个通道的均值和方差。
