一圖看懂Python生態(tài)圈圖像格式轉(zhuǎn)換

在Python生態(tài)圈里，最常用的圖像庫是PIL——盡管已經(jīng)被后來的pillow取代，但因為pillow的API幾乎完全繼承了PIL，所以大家還是約定俗成地稱其為PIL。除PIL之外，越來越多的程序員習(xí)慣使用openCV來處理圖像。另外，在GUI庫中，也有各自定義的圖像處理機(jī)制，比如wxPyton，定義了wx.Image做為圖像處理類，定義了wx.Bitmap做為圖像顯示類。

下圖梳理出了PIL讀寫圖像文件、cv2讀寫圖像文件、PIL對象和cv2對象互轉(zhuǎn)、PIL對象和wx.Image對象互轉(zhuǎn)、以及numpy數(shù)組轉(zhuǎn)存圖像的方法。掌握了這些方法，足可應(yīng)對各種各樣的圖像處理需求了。

1. PIL讀寫圖像文件

下面的代碼，演示了用PIL讀取png格式的圖像文件，剔除alpha通道后轉(zhuǎn)存為jpg格式的圖像文件。

>>> from PIL import Image
>>> im = Image.open(r'D:\CSDN\Python_Programming.png')
>>> r,g,b,a = im.split()
>>> im = Image.merge("RGB",(r,g,b))
>>> im.save(r'D:\CSDN\Python_Programming.jpg')

2. cv2讀寫圖像文件

下面的代碼，演示了用cv2讀取png格式的圖像文件，轉(zhuǎn)存為jpg格式的圖像文件。

>>> import cv2
>>> im = cv2.imread(r'D:\CSDN\Python_Programming.png')
>>> cv2.imwrite(r'D:\CSDN\Python_Programming.jpg', im)
True

3. PIL對象和cv2對象互轉(zhuǎn)

cv2格式的對象，本質(zhì)上就是numpy數(shù)組，也就是numpy.ndarray對象。只要能做到PIL對象和numpy數(shù)組互轉(zhuǎn)，自然就實現(xiàn)了PIL對象和cv2對象互轉(zhuǎn)。

下面的代碼，演示了用PIL讀取png格式的圖像文件，轉(zhuǎn)成numpy數(shù)組后保存為圖像文件。

>>> import cv2
>>> from PIL import Image
>>> import numpy as np
>>> im_pil = Image.open(r'D:\CSDN\Python_Programming.png')
>>> im_cv2 = np.array(im_pil)
>>> cv2.imwrite(r'D:\CSDN\Python_Programming.jpg', im_cv2)
True

下面的代碼，用cv2讀取png格式的圖像文件，轉(zhuǎn)成PIL對象后保存為圖像文件。

>>> import cv2
>>> from PIL import Image
>>> im_cv2 = cv2.imread(r'D:\CSDN\Python_Programming.png')
>>> im_pil = Image.fromarray(im_cv2)
>>> im_pil.save(r'D:\CSDN\Python_Programming.jpg')

4. PIL對象和wx.Image對象互轉(zhuǎn)

這是實現(xiàn)PIL對象和wx.Image對象互轉(zhuǎn)的兩個函數(shù)。

def PilImg2WxImg(pilImg):
    '''PIL的image轉(zhuǎn)化為wxImage'''
    image = wx.EmptyImage(pilImg.size[0],pilImg.size[1])
    image.SetData(pilImg.convert("RGB").tostring())
    image.SetAlphaData(pilImg.convert("RGBA").tostring()[3::4])
    return image

def WxImg2PilImg(wxImg):
    '''wxImage轉(zhuǎn)化為PIL的image'''
    pilImage = Image.new('RGB', (wxImg.GetWidth(), wxImg.GetHeight()))
    pilImage.fromstring(wxImg.GetData())

    if wxImg.HasAlpha():
        pilImage.convert( 'RGBA' )
        wxAlphaStr = wxImg.GetAlphaData()
        pilAlphaImage = Image.fromstring( 'L', (wxImg.GetWidth(), wxImg.GetHeight()), wxAlphaStr )
        pilImage.putalpha( pilAlphaImage )

    return pilImage

5. numpy數(shù)組轉(zhuǎn)存圖像

下面的代碼，生成了一張515x512像素的隨機(jī)圖像。

>>> from PIL import Image
>>> import numpy as np
>>> a = np.random.randint(0,256,((512,512,3)), dtype=np.uint8)
>>> im_pil = Image.fromarray(a)
>>> im_pil.save(r'D:\CSDN\random.jpg')