ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

使用python计算自己数据集的mean和std

2021-07-03 17:01:59  阅读:315  来源: 互联网

标签:std stdevs img means python print path os mean


首先先把数据集的图片路径保存在一个txt文件夹里面


   
  1. import os
  2. def generate(dir, label):
  3. listText = open( 'list.txt', 'a')
  4. for file in dir:
  5. fileType = os.path.split(file)
  6. if fileType[ 1] == '.txt':
  7. continue
  8. name = file + ' ' + str(int(label)) + '\n'
  9. listText.write(name)
  10. listText.close()
  11. outer_path = 'E:/lly/data/' # 这里是你的图片的目录
  12. if __name__ == '__main__':
  13. i = 1
  14. num = 0
  15. personlist = os.listdir(outer_path) # 列举文件夹
  16. personlist.sort()
  17. for person in personlist:
  18. personPath = outer_path+person + "/"
  19. fingerlist = os.listdir(personPath)
  20. fingerlist.sort()
  21. for finger in fingerlist:
  22. finallPATH=os.path.join(outer_path, person,finger)
  23. finallPATH=finallPATH.replace( '\\', '/')
  24. listText = open( 'image_list.txt', 'a')
  25. fileType = os.path.split(finallPATH)
  26. name = finallPATH+ '\n'
  27. listText.write(name)
  28. listText.close()
  29. i += 1

计算自己数据集的均值和方差:


   
  1. # -*- coding: utf-8 -*-**
  2. import numpy as np
  3. import cv2
  4. import random
  5. import os
  6. # calculate means and std 注意换行\n符号**
  7. # train.txt中每一行是图像的位置信息**
  8. path = 'C:/Users/lenovo/PycharmProjects/my/image_list.txt'
  9. means = [ 0, 0, 0]
  10. stdevs = [ 0, 0, 0]
  11. index = 1
  12. num_imgs = 0
  13. with open(path, 'r') as f:
  14. lines = f.readlines()
  15. # random.shuffle(lines)
  16. for line in lines:
  17. print(line)
  18. print( '{}/{}'.format(index, len(lines)))
  19. index += 1
  20. a = os.path.join(line)
  21. # print(a[:-1])
  22. num_imgs += 1
  23. img = cv2.imread(a[: -1])
  24. print(img, 22)
  25. img = np.asarray(img)
  26. img = img.astype(np.float32) / 255.
  27. for i in range( 3):
  28. means[i] += img[:, :, i].mean()
  29. stdevs[i] += img[:, :, i].std()
  30. print(num_imgs)
  31. means.reverse()
  32. stdevs.reverse()
  33. means = np.asarray(means) / num_imgs
  34. stdevs = np.asarray(stdevs) / num_imgs
  35. print( "normMean = {}".format(means))
  36. print( "normStd = {}".format(stdevs))
  37. print( 'transforms.Normalize(normMean = {}, normStd = {})'.format(means, stdevs))

 

参考:https://blog.csdn.net/weixin_38533896/article/details/85951903

标签:std,stdevs,img,means,python,print,path,os,mean
来源: https://blog.csdn.net/weixin_43955293/article/details/118441891

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有