计算pytorch标准化(Normalize)所需要数据集的均值和方差实例


Posted in Python onJanuary 15, 2020

pytorch做标准化利用transforms.Normalize(mean_vals, std_vals),其中常用数据集的均值方差有:

if 'coco' in args.dataset:
  mean_vals = [0.471, 0.448, 0.408]
  std_vals = [0.234, 0.239, 0.242]
elif 'imagenet' in args.dataset:
  mean_vals = [0.485, 0.456, 0.406]
  std_vals = [0.229, 0.224, 0.225]

计算自己数据集图像像素的均值方差:

import numpy as np
import cv2
import random
 
# calculate means and std
train_txt_path = './train_val_list.txt'
 
CNum = 10000   # 挑选多少图片进行计算
 
img_h, img_w = 32, 32
imgs = np.zeros([img_w, img_h, 3, 1])
means, stdevs = [], []
 
with open(train_txt_path, 'r') as f:
  lines = f.readlines()
  random.shuffle(lines)  # shuffle , 随机挑选图片
 
  for i in tqdm_notebook(range(CNum)):
    img_path = os.path.join('./train', lines[i].rstrip().split()[0])
 
    img = cv2.imread(img_path)
    img = cv2.resize(img, (img_h, img_w))
    img = img[:, :, :, np.newaxis]
    
    imgs = np.concatenate((imgs, img), axis=3)
#     print(i)
 
imgs = imgs.astype(np.float32)/255.
 
 
for i in tqdm_notebook(range(3)):
  pixels = imgs[:,:,i,:].ravel() # 拉成一行
  means.append(np.mean(pixels))
  stdevs.append(np.std(pixels))
 
# cv2 读取的图像格式为BGR,PIL/Skimage读取到的都是RGB不用转
means.reverse() # BGR --> RGB
stdevs.reverse()
 
print("normMean = {}".format(means))
print("normStd = {}".format(stdevs))
print('transforms.Normalize(normMean = {}, normStd = {})'.format(means, stdevs))

以上这篇计算pytorch标准化(Normalize)所需要数据集的均值和方差实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python中的二叉树查找算法模块使用指南
Jul 04 Python
python中Genarator函数用法分析
Apr 08 Python
简单实现python爬虫功能
Dec 31 Python
一个基于flask的web应用诞生 flask和mysql相连(4)
Apr 11 Python
Pycharm学习教程(3) 代码运行调试
May 03 Python
pandas DataFrame实现几列数据合并成为新的一列方法
Jun 08 Python
Python import与from import使用及区别介绍
Sep 06 Python
python学生管理系统
Jan 30 Python
pandas数据筛选和csv操作的实现方法
Jul 02 Python
Python爬虫实现使用beautifulSoup4爬取名言网功能案例
Sep 15 Python
wxPython实现带颜色的进度条
Nov 19 Python
PyTorch中permute的用法详解
Dec 30 Python
pytorch 图像中的数据预处理和批标准化实例
Jan 15 #Python
pytorch实现特殊的Module--Sqeuential三种写法
Jan 15 #Python
python实现删除列表中某个元素的3种方法
Jan 15 #Python
python opencv根据颜色进行目标检测的方法示例
Jan 15 #Python
Python基于Tensor FLow的图像处理操作详解
Jan 15 #Python
OpenCV哈里斯(Harris)角点检测的实现
Jan 15 #Python
Pytorch模型转onnx模型实例
Jan 15 #Python
You might like
全国FM电台频率大全 - 29 青海省
2020/03/11 无线电
用PHP的ob_start();控制您的浏览器cache!
2006/11/25 PHP
php学习笔记 数组遍历实现代码
2011/06/09 PHP
使用Smarty 获取当前日期时间和格式化日期时间的方法详解
2013/06/18 PHP
php将mysql数据库整库导出生成sql文件的具体实现
2014/01/08 PHP
php实现指定字符串中查找子字符串的方法
2015/03/17 PHP
PHP实现简单实用的分页类代码
2016/04/08 PHP
Extjs学习笔记之四 工具栏和菜单
2010/01/07 Javascript
js渐变显示渐变消失示例代码
2013/08/01 Javascript
jquery隐藏标签和显示标签的实例
2013/11/11 Javascript
JavaScript截取字符串的2个函数介绍
2014/08/27 Javascript
jQuery实现仿路边灯箱广告图片轮播效果
2015/04/15 Javascript
用jmSlip编写移动端顶部日历选择控件
2016/10/24 Javascript
jquery.form.js异步提交表单详解
2017/04/25 jQuery
Bootstrap4如何定制自己的颜色和风格
2018/02/26 Javascript
详解Vue单元测试case写法
2018/05/24 Javascript
Vue实现远程获取路由与页面刷新导致404错误的解决
2019/01/31 Javascript
微信小程序实现左右列表联动
2020/05/19 Javascript
微信小程序实现Session功能及无法获取session问题的解决方法
2019/05/07 Javascript
这15个Vue指令,让你的项目开发爽到爆
2019/10/11 Javascript
vue-cli创建的项目中的gitHooks原理解析
2020/02/14 Javascript
解决VUE自定义拖拽指令时 onmouseup 与 click事件冲突问题
2020/07/24 Javascript
[42:50]NB vs VP 2018国际邀请赛小组赛BO2 第二场 8.18
2018/08/19 DOTA
Python 性能优化技巧总结
2016/11/01 Python
linux环境下python中MySQLdb模块的安装方法
2017/06/16 Python
python九九乘法表的实例
2017/09/26 Python
Python中如何将一个类方法变为多个方法
2019/12/30 Python
tensorflow 环境变量设置方式
2020/02/06 Python
python中使用you-get库批量在线下载bilibili视频的教程
2020/03/10 Python
Python 获取异常(Exception)信息的几种方法
2020/12/29 Python
2015年清明节演讲稿范文
2015/03/17 职场文书
2015年党小组工作总结
2015/05/26 职场文书
干货:我将这样书写我的演讲稿!
2019/05/09 职场文书
大学生奶茶店创业计划书
2019/06/25 职场文书
Django rest framework如何自定义用户表
2021/06/09 Python
Redis超详细讲解高可用主从复制基础与哨兵模式方案
2022/04/07 Redis