计算图像数据集的RGB均值

最新推荐文章于 2023-05-08 20:12:47 发布

转载

最新推荐文章于 2023-05-08 20:12:47 发布 · 2k 阅读

4 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/Terrypython/p/11319237.html

文章标签：

#人工智能 #python #java

这篇博客介绍了如何计算图像数据集的RGB均值。作者通过编写程序实现了这一功能，该程序适用于人工智能领域的图像处理任务。文章内容包括程序代码展示和简单的目录结构，提供了从Python或Java角度解决这一问题的思路。

最近在跑代码的时候，需要用到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.
  """
  flower_root = os.path.join(dataset_dir, 'Stripe')
  directories = [