from gevent import monkey monkey.patch_all()
import gevent
from haishoku.haishoku import Haishoku
import math
from colorsys import rgb_to_hsv
import os
from collections import OrderedDict
import pandas as pd
import time from openpyxl import Workbook
from openpyxl.styles import PatternFill, fills colors = dict((
((255, 182, 193), "浅粉色"),
((255, 192, 203), "粉红"),
((220, 20, 60), "猩红"),
((255, 240, 245), "脸红的淡紫色"),
((219, 112, 147), "苍白的紫罗兰红色"),
((255, 105, 180), "热情的粉红"),
((255, 20, 147), "深粉色"),
((199, 21, 133), "适中的紫罗兰红色"),
((218, 112, 214), "兰花的紫色"),
((216, 191, 216), "蓟"),
((221, 160, 221), "李子"),
((238, 130, 238), "紫罗兰"),
((255, 0, 255), "洋红"),
((139, 0, 139), "深洋红色"),
((128, 0, 128), "紫色"),
((186, 85, 211), "适中的兰花紫"),
((148, 0, 211), "深紫罗兰色"),
((153, 50, 204), "深兰花紫"),
((75, 0, 130), "靛青"),
((138, 43, 226), "深紫罗兰的蓝色"),
((147, 112, 219), "适中的紫色"),
((123, 104, 238), "适中的板岩暗蓝灰色"),
((106, 90, 205), "板岩暗蓝灰色"),
((72, 61, 139), "深岩暗蓝灰色"),
((230, 230, 250), "薰衣草花的淡紫色"),
((248, 248, 255), "幽灵的白色"),
((0, 0, 255), "纯蓝"),
((0, 0, 205), "适中的蓝色"),
((25, 25, 112), "午夜的蓝色"),
((0, 0, 139), "深蓝色"),
((0, 0, 128), "海军蓝"),
((65, 105, 225), "宝蓝"),
((100, 149, 237), "矢车菊的蓝色"),
((176, 196, 222), "淡钢蓝"),
((119, 136, 153), "浅石板灰"),
((112, 128, 144), "石板灰"),
((30, 144, 255), "道奇蓝"),
((240, 248, 255), "爱丽丝蓝"),
((70, 130, 180), "钢蓝"),
((135, 206, 250), "淡蓝色"),
((135, 206, 235), "天蓝色"),
((0, 191, 255), "深天蓝"),
((173, 216, 230), "淡蓝"),
((176, 224, 230), "火药蓝"),
((95, 158, 160), "军校蓝"),
((240, 255, 255), "蔚蓝色"),
((225, 255, 255), "淡青色"),
((175, 238, 238), "苍白的绿宝石"),
((0, 255, 255), "水绿色"),
((0, 206, 209), "深绿宝石"),
((47, 79, 79), "深石板灰"),
((0, 139, 139), "深青色"),
((0, 128, 128), "水鸭色"),
((72, 209, 204), "适中的绿宝石"),
((32, 178, 170), "浅海洋绿"),
((64, 224, 208), "绿宝石"),
((127, 255, 170), "绿玉"),
((0, 250, 154), "适中的碧绿色"),
((245, 255, 250), "适中的春天的绿色"),
((0, 255, 127), "薄荷奶油"),
((60, 179, 113), "春天的绿色"),
((46, 139, 87), "海洋绿"),
((240, 255, 0), "蜂蜜"),
((144, 238, 144), "淡绿色"),
((152, 251, 152), "苍白的绿色"),
((143, 188, 143), "深海洋绿"),
((50, 205, 50), "酸橙绿"),
((0, 255, 0), "酸橙色"),
((34, 139, 34), "森林绿"),
((0, 128, 0), "纯绿"),
((0, 100, 0), "深绿色"),
((127, 255, 0), "查特酒绿"),
((124, 252, 0), "草坪绿"),
((173, 255, 47), "绿黄色"),
((85, 107, 47), "橄榄土褐色"),
((107, 142, 35), "米色（浅褐色）"),
((250, 250, 210), "浅秋麒麟黄"),
((255, 255, 240), "象牙色"),
((255, 255, 224), "浅黄色"),
((255, 255, 0), "纯黄"),
((128, 128, 0), "橄榄"),
((189, 183, 107), "深卡其布"),
((255, 250, 205), "柠檬薄纱"),
((238, 232, 170), "灰秋麒麟"),
((240, 230, 140), "卡其布"),
((255, 215, 0), "金"),
((255, 248, 220), "玉米色"),
((218, 165, 32), "秋麒麟"),
((255, 250, 240), "花的白色"),
((253, 245, 230), "老饰带"),
((245, 222, 179), "小麦色"),
((255, 228, 181), "鹿皮鞋"),
((255, 165, 0), "橙色"),
((255, 239, 213), "番木瓜"),
((255, 235, 205), "漂白的杏仁"),
((255, 222, 173), "Navajo白"),
((250, 235, 215), "古代的白色"),
((210, 180, 140), "晒黑"),
((222, 184, 135), "结实的树"),
((255, 228, 196), "（浓汤）乳脂，番茄等"),
((255, 140, 0), "深橙色"),
((250, 240, 230), "亚麻布"),
((205, 133, 63), "秘鲁"),
((255, 218, 185), "桃色"),
((244, 164, 96), "沙棕色"),
((210, 105, 30), "巧克力"),
((139, 69, 19), "马鞍棕色"),
((255, 245, 238), "海贝壳"),
((160, 82, 45), "黄土赭色"),
((255, 160, 122), "浅鲜肉（鲑鱼）色"),
((255, 127, 80), "珊瑚"),
((255, 69, 0), "橙红色"),
((233, 150, 122), "深鲜肉（鲑鱼）色"),
((255, 99, 71), "番茄"),
((255, 228, 225), "薄雾玫瑰"),
((250, 128, 114), "鲜肉（鲑鱼）色"),
((255, 250, 250), "雪"),
((240, 128, 128), "淡珊瑚色"),
((188, 143, 143), "玫瑰棕色"),
((205, 92, 92), "印度红"),
((255, 0, 0), "纯红"),
((165, 42, 42), "棕色"),
((178, 34, 34), "耐火砖"),
((139, 0, 0), "深红色"),
((128, 0, 0), "栗色"),
((255, 255, 255), "纯白"),
((245, 245, 245), "白烟"),
((220, 220, 220), "Gainsboro"),
((211, 211, 211), "浅灰色"),
((192, 192, 192), "银白色"),
((169, 169, 169), "深灰色"),
((128, 128, 128), "灰色"),
((105, 105, 105), "暗淡的灰色"),
((0, 0, 0), "纯黑")
)) def to_hsv(color):
""" converts color tuples to floats and then to hsv """
return rgb_to_hsv(*[x / 255.0 for x in color]) # rgb_to_hsv wants floats! def color_dist(c1, c2):
""" returns the squared euklidian distance between two color vectors in hsv space """
return sum((a - b) ** 2 for a, b in zip(to_hsv(c1), to_hsv(c2))) def min_color_diff(color_to_match, colors):
""" returns the `(distance, color_name)` with the minimal distance to `colors`"""
return min( # overal best is the best match to any color:
(color_dist(color_to_match, test), colors[test]) # (distance to `test` color, color name)
for test in colors) def rgb2hex(rgbcolor):
r, g, b = rgbcolor
color = "#"
color += str(hex(r)).replace('x', '')[-2:]
color += str(hex(g)).replace('x', '')[-2:]
color += str(hex(b)).replace('x', '')[-2:]
return color # 新建一个新的工作表（未保存）。
wb = Workbook()
# 保存文件，若加载路径与保存的路径一致将会被覆盖
my_sheet = wb.worksheets[0]
row_title = ["图片路径名称","主要图片颜色","主要图片颜色hex","主要图片颜色占比","次要图片颜色","次要图片颜色hex","次要图片颜色占比","次次要图片颜色","次次要图片颜色hex","次次要图片颜色占比"]
my_sheet.append(row_title) def get_colorname(index,path):
single_data = OrderedDict()
try:
haishoku = Haishoku.loadHaishoku(path)
# single_data['图片路径名称'] = path
palette = haishoku.palette
main_color = palette[0][1]
main_color_pct = palette[0][0]
mian_colorname = min_color_diff(main_color, colors)[1]
tmp_main_color = rgb2hex(main_color)[1:]
# single_data['主要图片颜色'] = mian_colorname
# single_data['主要图片颜色rgb'] = rgb2hex(main_color)
# single_data['主要图片颜色占比'] = main_color_pct
second_color = palette[1][1]
second_color_pct = palette[1][0]
second_colorname = min_color_diff(second_color, colors)[1]
tmp_second_color = rgb2hex(second_color)[1:]
# single_data['次要图片颜色'] = second_colorname
# single_data['次要图片颜色rgb'] = rgb2hex(second_color)
# single_data['次要图片颜色占比'] = second_color_pct
thred_color = palette[2][1]
thred_color_pct = palette[2][0]
thred_colorname = min_color_diff(thred_color, colors)[1]
tmp_thred_color = rgb2hex(thred_color)[1:]
# single_data['次次要图片颜色'] = thred_colorname
# single_data['次次要图片颜色rgb'] = rgb2hex(thred_color)
# single_data['次次要图片颜色占比'] = thred_color_pct
row_line = [path,mian_colorname,"#"+tmp_main_color,main_color_pct,second_colorname,"#"+tmp_second_color,second_color_pct,thred_colorname,"#"+tmp_thred_color,thred_color_pct]
my_sheet.append(row_line)
my_sheet["C"+str(2+int(index))].fill = PatternFill(fill_type=fills.FILL_SOLID, fgColor=tmp_main_color, bgColor=tmp_main_color)
my_sheet["F"+str(2+int(index))].fill = PatternFill(fill_type=fills.FILL_SOLID, fgColor=tmp_second_color, bgColor=tmp_second_color)
my_sheet["I"+str(2+int(index))].fill = PatternFill(fill_type=fills.FILL_SOLID, fgColor=tmp_thred_color, bgColor=tmp_thred_color)
print(index+1)
# print(path,'主要颜色是：'+mian_colorname,'主要颜色占比：'+str(main_color_pct*100)+'%',' 次要颜色是：'+second_colorname,'次要颜色占比：'+str(second_color_pct*100)+'%',' 次次要颜色是：'+thred_colorname,'次次要颜色占比：'+str(thred_color_pct*100)+'%')
except Exception as e:
print('错误-->', path, e) pic_names = os.listdir('./picture')
all_abs_picname = (os.path.join('./picture', name) for name in pic_names if name != '.DS_Store' and len(name)>4) greenlets = [gevent.spawn(get_colorname, index,path) for index,path in enumerate(all_abs_picname)]
gevent.joinall(greenlets) if not os.path.exists('图片颜色导出'):
os.mkdir(os.path.join(os.getcwd(), '图片颜色导出')) wb.save(r'图片颜色导出' + '/' + time.strftime("%Y%m%d%H%M") + '.xlsx')
print('done')