利用python編寫(xiě)一個(gè)圖片主色轉(zhuǎn)換的腳本

更新時(shí)間：2017年12月07日 08:40:28 作者：WHC

這篇文章主要給大家介紹了關(guān)于利用python編寫(xiě)一個(gè)圖片主色轉(zhuǎn)換腳本的相關(guān)資料，主要使用的是Python中的Pillow圖像處理庫(kù)，文中通過(guò)示例代碼介紹的非常詳細(xì)，需要的朋友可以參考借鑒，下面隨著小編來(lái)一起看看吧。

前言

最近由于項(xiàng)目特需老是替換主題顏色，同時(shí)app里一些資源icon圖片主色也要改，美工不提供切圖只能靠自己了，開(kāi)始想在iconfont上面找但是數(shù)量比較多太浪費(fèi)時(shí)間，然后就想到python的Pillow在圖像處理方便很強(qiáng)大。

Pillow是Python里的圖像處理庫(kù)（PIL：Python Image Library），提供了了廣泛的文件格式支持，強(qiáng)大的圖像處理能力，主要包括圖像儲(chǔ)存、圖像顯示、格式轉(zhuǎn)換以及基本的圖像處理操作等。

本文就編寫(xiě)了一個(gè)批量替換圖片主色的腳本changeImageColor.py。

實(shí)現(xiàn)思路

1.pip 安裝Pillow庫(kù)引入Image類(lèi)

2.在當(dāng)前目錄下創(chuàng)建存放轉(zhuǎn)換后圖片目錄

3.獲取當(dāng)前目錄路徑，以及圖片文件

4.遍歷所有圖片文件并創(chuàng)建對(duì)應(yīng)Image對(duì)象

5.獲取Image對(duì)象背景顏色rgba值

6.遍歷Image對(duì)象所有像素點(diǎn)

7.把不是背景像素點(diǎn)顏色替換為要轉(zhuǎn)換輸入顏色值

8.保存Image對(duì)象到新目錄下面

代碼實(shí)現(xiàn)

from PIL import Image
import os
class ChangeImageColor(object):
  @classmethod
  def startHandle(self, rgb):
    # 獲取當(dāng)前路徑，并創(chuàng)建新目錄用于輸出結(jié)果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) # 當(dāng)前目錄路徑
      print('dirs: ', dirs) # 當(dāng)前路徑下所有子目錄
      print('files: ', files) # 當(dāng)前路徑下所有非目錄子文件      
      # 遍歷下所有圖片文件
      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每個(gè)像素
            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進(jìn)制轉(zhuǎn)換為rgb
  @classmethod
  def hex2rgb(self, hexcolor):
    rgb = ((hexcolor >> 16) & 0xff,
        (hexcolor >> 8) & 0xff,
        hexcolor & 0xff
        )
    return rgb
if __name__ == '__main__':
  hexColor = int(input('請(qǐng)輸入新16進(jìn)制顏色值:'), 16)
  ChangeImageColor.startHandle(ChangeImageColor.hex2rgb(hexColor))

演示