import os
import cv2
import numpy as np
path = r'./JPEGImages' #图片保存路径
def compute(path):
file_names = os.listdir(path)
per_image_Rmean = []
per_image_Gmean = []
per_image_Bmean = []
for file_name in file_names:
img = cv2.imread(os.path.join(path, file_name), 1)
per_image_Bmean.append(np.mean(img[:, :, 0]))
per_image_Gmean.append(np.mean(img[:, :, 1]))
per_image_Rmean.append(np.mean(img[:, :, 2]))
R_mean = np.mean(per_image_Rmean)
G_mean = np.mean(per_image_Gmean)
B_mean = np.mean(per_image_Bmean)
stdR = np.std(per_image_Rmean)
stdG = np.std(per_image_Gmean)
stdB = np.std(per_image_Bmean)
return R_mean, G_mean, B_mean, stdR, stdG, stdB
if __name__ == '__main__':
R, G, B, stdR, stdG, stdB = compute(path)
print("R= ", R, "G= ", G, "B=", B, "stdR = ", stdR, "stdG = ", stdG, "stdB =", stdB)
pytrhon计算图片数据集均值和方差
最新推荐文章于 2024-04-24 23:43:15 发布