将所有的图片根据亮度值分成4个区间,能看出不同的亮度值对模型的影响。比如有的论文按80,130将图片分为Low,Middle和High。
from PIL import Image, ImageStat
import math
import os
import matplotlib.pyplot as plt
import numpy as np
import cv2
import shutil
def brightness(im_file):
im = Image.open(im_file)
stat = ImageStat.Stat(im)
r, g, b = stat.mean
return math.sqrt(0.241 * (r ** 2) + 0.691 * (g ** 2) + 0.068 * (b ** 2))
# def illumination_averge(directory_name):
# illumination_averge = []
# for filename in os.listdir(directory_name):
# # 用os库读取文件夹内所有图片名称以操作
# #