1. 程式人生 > 程式設計 >Python實現轉換圖片背景顏色程式碼

Python實現轉換圖片背景顏色程式碼

在使用圖片的時候有時候我們希望改變背景顏色,這樣就只關注於圖片本身。比如在連連看中就只有圖片,而沒有背景,其實我個人感覺有背景好看一點。

兩種方法,思路一致:

法一:

import cv2

# 修改透明背景為白色
def transparence2white(img):
  sp=img.shape # 獲取圖片維度
  width=sp[0] # 寬度
  height=sp[1] # 高度
  for yh in range(height):
    for xw in range(width):
      color_d=img[xw,yh] # 遍歷影象每一個點,獲取到每個點4通道的顏色資料
      if(color_d[3]==0): # 最後一個通道為透明度,如果其值為0,即影象是透明
        img[xw,yh]=[255,255,255] # 則將當前點的顏色設定為白色,且影象設定為不透明
  return img

img=cv2.imread('bar.png',-1) # 讀取圖片。-1將圖片透明度傳入,資料由RGB的3通道變成4通道
img=transparence2white(img) # 將圖片傳入,改變背景色後,返回
cv2.imwrite('bar.png',img) # 儲存圖片,檔名自定義,也可以覆蓋原檔案

法二:

from PIL import Image
 
def transparence2white(img):
#   img=img.convert('RGBA') # 此步驟是將影象轉為灰度(RGBA表示4x8位畫素,帶透明度掩模的真彩色;CMYK為4x8位畫素,分色等),可以省略
  sp=img.size
  width=sp[0]
  height=sp[1]
  print(sp)
  for yh in range(height):
    for xw in range(width):
      dot=(xw,yh)
      color_d=img.getpixel(dot) # 與cv2不同的是,這裡需要用getpixel方法來獲取維度資料
      if(color_d[3]==0):
        color_d=(255,255)
        img.putpixel(dot,color_d) # 賦值的方法是通過putpixel
  return img
 

img=Image.open('bar.png')
img=transparence2white(img)
# img.show() # 顯示圖片
img.save('bar3.png') # 儲存圖片

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。