以下是一个 Python 函数,可用于在指定文件夹下查找给定颜色范围内的所有符合条件的图片
import os
from PIL import Image
def find_color_images(color, folder_path, tolerance=50):
"""
在指定文件夹下查找给定颜色范围内的所有符合条件的图片
:param color: tuple,(r, g, b) 形式的颜色值
:param folder_path: 待查找图片的文件夹路径
:param tolerance: 颜色容差,取值范围为 0~255,默认为 50
:return: 符合条件的图片名称列表
"""
file_list = os.listdir(folder_path)
result_list = []
for file_name in file_list:
if file_name.endswith('.jpg') or file_name.endswith('.png'):
image_path = os.path.join(folder_path, file_name)
with Image.open(image_path) as img:
width, height = img.size
color_count = 0
for x in range(width):
for y in range(height):
pixel_color = img.getpixel((x, y))
if (abs(pixel_color[0] - color[0]) <= tolerance and
abs(pixel_color[1] - color[1]) <= tolerance and
abs(pixel_color[2] - color[2]) <= tolerance):
color_count += 1
if color_count / (width * height) > 0.01:
result_list.append(file_name)
return result_list
该函数使用了 os 和 PIL 库。首先,通过传入的文件夹路径获取该文件夹下的所有文件名,然后遍历每个文件名,判断是否为 JPG 或 PNG 格式的图片。若是,则打开该图片,遍历其中所有像素点,若该像素点的颜色与目标颜色在三个通道上的差值均不大于设定的容差值,则认为该像素点属于目标颜色,计数器加一。
统计完整张图片中目标颜色的像素点个数后,将其占比与设定的阈值(此处为 0.01)比较,若大于该阈值,则将该图片名添加到结果列表中。最终返回符合条件的文件名列表。
需要注意的是,该函数只针对单一颜色进行查找,若需要同时查找多种颜色范围,可以在函数外部加一个循环,依次传入不同的颜色值。