用PYTHON实现按键精灵找图着色实战源码

以下是一个 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)比较,若大于该阈值,则将该图片名添加到结果列表中。最终返回符合条件的文件名列表。

需要注意的是,该函数只针对单一颜色进行查找,若需要同时查找多种颜色范围,可以在函数外部加一个循环,依次传入不同的颜色值。

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

爱分享的大户爱

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值