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


            前言

最近由于项目特需老是替换主题颜色，同时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 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))



演示



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



python真的很不错^_^。

总结

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