计算图像数据集的RGB均值

最近在跑代码的时候，需要用到RGB三个通道上的均值，如下图所示：

写了一个程序，如下：

import os
import cv2
import random
import numpy as np
 
#数据集的位置
dataset_dir = 'data'

# 验证集图像数量
_NUM_VALIDATION = 1000

# 随机种子
_RANDOM_SEED = 0

def _get_filenames_and_classes(dataset_dir):
  """Returns a list of filenames and inferred class names.

  Args:
    dataset_dir:A directory containing a set of different datasets.Each dataset
    contains a set of subdirectories representing class names.Each subdirectory 
    should contain PNG or JPG encoded images.
  Returns:
    A list of image file paths, relative to `dataset_dir` and the list of
    subdirectories, representing class names