You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

如何用Python的ColorThief批量获取多张图片的主色调?

批量提取图片主色调的实现方案

当然可以批量处理,以下是直接可用的代码,能遍历你桌面文件夹里的图片,提取每张的RGB主色调,并输出文件名与对应RGB值的结果,还能处理异常图片(比如损坏文件):

from colorthief import ColorThief
import os

# 桌面路径,跨平台兼容
desktop_path = os.path.expanduser("~/Desktop")
# 支持的图片格式,可按需添加
supported_formats = ('.jpg', '.jpeg', '.png', '.bmp', '.gif')

# 存储结果的列表
results = []

# 遍历桌面文件夹
for filename in os.listdir(desktop_path):
    # 过滤出图片文件
    if filename.lower().endswith(supported_formats):
        file_path = os.path.join(desktop_path, filename)
        try:
            # 提取主色调
            color_thief = ColorThief(file_path)
            dominant_color = color_thief.get_color(quality=1)
            results.append((filename, dominant_color))
        except Exception as e:
            # 处理异常情况,比如损坏的图片
            print(f"处理文件 {filename} 时出错: {str(e)}")

# 打印结果(两列格式)
print("文件名\t\tRGB主色调")
print("------------------------------")
for name, rgb in results:
    print(f"{name}\t{rgb}")

# 可选:将结果保存为CSV文件,方便用表格软件查看
import csv
with open(os.path.join(desktop_path, "图片主色调结果.csv"), 'w', newline='', encoding='utf-8') as f:
    writer = csv.writer(f)
    writer.writerow(["文件名", "RGB主色调"])
    writer.writerows(results)
print("结果已保存到桌面的「图片主色调结果.csv」文件")

说明:

  • 代码会自动识别桌面下的常见格式图片,若有其他格式可自行添加到supported_formats中
  • 加入了异常处理,遇到损坏或无法读取的图片会跳过并提示错误
  • 除了控制台打印两列结果,还会在桌面生成CSV文件,用Excel或WPS打开就能看到规整的表格

内容的提问来源于stack exchange,提问作者JC91

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.08.01 07:02:33