Python PIL方式打開的圖片判斷維度

1. PIL方式打開的圖片判斷維度

　　   好久沒更新啦，哈哈哈~~!今天跟寶寶們分享一篇如何判斷灰度圖像和彩色圖像維度的方法。我們在讀取灰度圖像和彩色圖像時，發現讀取出來的圖片維度不同，當我們要做后續操作的時候，很可能去人工判斷，那樣顯得自己憨憨的，今天就幫各位寶寶解決這個問題啦！！！

　　讀取灰度圖片，獲取的格式是：（height, width），此時就只有兩維了，灰度圖像channel為1，就會被省略掉。讀取彩色圖片，獲取的格式是：（height，width，channel），此時channel一般為3，會被保留下來。那么問題來了，在Python里面好像還沒提供直接獲取維度的方法，最多提供了Image.shape，單這就只能獲取上訴說的（height, width）或（height，width，channel）這種格式信息，而不是直接得到一個2或者3。

　　好啦，不多說啦，直接上code......

2. 例子

2.1 code

import numpy as np
from PIL import Image
from torchvision import transforms


def get_Image_dim_len(image_dir):
    image = Image.open(image_dir)
    image_dim_len = len(np.array(image).shape)
    image.show()
    print("The dim of Image: ", image_dim_len)



if __name__ == '__main__':
    # 彩色圖像
    image_color = './test_datasets/1.jpg'
    get_Image_dim_len(image_color)
    # # 灰度圖像
    # image_gray = './test_datasets/2.jpg'
    # get_Image_dim_len(image_gray)

2.2 結果展示

 

              （a）彩色圖像                             （b）灰度圖像

圖(a)的輸出為：The dim of Image:  3

圖(b)的輸出為：The dim of Image:  2

3. 總結

　　努力去愛周圍的每一個人，付出，不一定有收獲，但是不付出就一定沒有收獲！ 給街頭賣藝的人零錢，不和深夜還在擺攤的小販討價還價。願我的博客對你有所幫助(*^▽^*)(*^▽^*)！

　  如果客官喜歡小生的園子，記得關注小生喲，小生會持續更新(#^.^#)(#^.^#)。