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

时间:2021-05-22

前言

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

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

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

实现思路

1.pip 安装Pillow库引入Image类

2.在当前目录下创建存放转换后图片目录

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

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

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

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

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

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

代码实现

from PIL import Imageimport osclass 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 rgbif __name__ == '__main__': hexColor = int(input('请输入新16进制颜色值:'), 16) ChangeImageColor.startHandle(ChangeImageColor.hex2rgb(hexColor))

演示

转换结果result 目录下就是转换新图片

python真的很不错^_^。

总结

以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,如果有疑问大家可以留言交流,谢谢大家对的支持。

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章