利用python编写一个图片主色转换的脚本

更新时间：2020-05-13 08:30:01 作者：startmvc

前言最近由于项目特需老是替换主题颜色，同时app里一些资源icon图片主色也要改，美工不

前言

最近由于项目特需老是替换主题颜色，同时app里一些资源icon图片主色也要改，美工不提供切图只能靠自己了，开始想在iconfont上面找但是数量比较多太浪费时间，然后就想到python的Pillow在图像处理方便很强大。

Pillow是Python里的图像处理库（PIL：Python Image Library），提供了了广泛的文件格式支持，强大的图像处理能力，主要包括图像储存、图像显示、格式转换以及基本的图像处理操作等。

本文就编写了一个批量替换图片主色的脚本changeImageColor.py。

实现思路

1.pip 安装Pillow库引入Image类

3.获取当前目录路径，以及图片文件

4.遍历所有图片文件并创建对应Image对象

5.获取Image对象背景颜色rgba值

6.遍历Image对象所有像素点

7.把不是背景像素点颜色替换为要转换输入颜色值

8.保存Image对象到新目录下面

代码实现


from PIL import Image
import os
class ChangeImageColor(object):
 @classmethod
 def startHandle(self, rgb):
 # 获取当前路径，并创建新目录用于输出结果image
 path = os.getcwd() + '/images/res'
 npath = os.getcwd() + '/images/res/result/'
 if not os.path.exists(npath):
 os.makedirs(npath)
 else:
 # 如果存在相同新目录那么删除下面文件
 for root, dirs, files in os.walk(npath):
 for file_name in files:
 os.remove(npath + file_name) 
 # 新颜色值
 nr,ng,nb = rgb
 # 存放背景颜色
 br,bg,bb, ba = 0, 0, 0, 0
 # 遍历目录
 for root, dirs, files in os.walk(path):
 print('root: ', root) # 当前目录路径
 print('dirs: ', dirs) # 当前路径下所有子目录
 print('files: ', files) # 当前路径下所有非目录子文件 
 # 遍历下所有图片文件
 for file_name in files:
 if file_name != '.DS_Store':
 image = Image.open(root + '/' + file_name)
 if image is not None:
 image_width, image_height = image.size
 # 遍历Image每个像素
 for i in range(image_width):
 for j in range(image_height):
 xy = (i,j)
 # 下面是获取像素和比较像素
 color = image.getpixel(xy)
 color_num = len(color)
 # 判断颜色是否有alpha值
 if color_num == 4:
 r, g, b, a = color
 if i == 0 and j == 0:
 br, bg, bb, ba = color
 if br != r or bg != g or bb != b:
 # 替换像素并保留alpha值
 image.putpixel(xy, (nr, ng, nb,a))
 elif color_num == 3:
 r, g, b = color
 if i == 0 and j == 0:
 br, bg, bb = color
 if br != r or bg != g or bb != b: image.putpixel(xy, (nr, ng, nb))

 image.save(npath + file_name)
 # 把16进制转换为rgb
 @classmethod
 def hex2rgb(self, hexcolor):
 rgb = ((hexcolor >> 16) & 0xff,
 (hexcolor >> 8) & 0xff,
 hexcolor & 0xff
 )
 return rgb
if __name__ == '__main__':
 hexColor = int(input('请输入新16进制颜色值:'), 16)
 ChangeImageColor.startHandle(ChangeImageColor.hex2rgb(hexColor))

演示