專案下的image資料夾裡有五張.bmp影象

 1 import os
 2 import cv2
 3 
 4 def get_img_list(dir, firelist, ext=None):
 5     newdir = dir
 6     if os.path.isfile(dir):  # 如果是檔案
 7         if ext is None:
 8             firelist.append(dir)
 9         elif ext in dir[-3:]:
10             firelist.append(dir)
11     elif
 
 os.path.isdir(dir):  # 如果是目錄
12         for s in os.listdir(dir):
13             newdir = os.path.join(dir, s)
14             get_img_list(newdir, firelist, ext)
15 
16     return firelist
17 
18 def read_img():
19     image_path = './image'
20     imglist = get_img_list(image_path, [], 'bmp')
21     print
 
(imglist.__len__())
22     for imgpath in imglist:
23         # print(imgpath)
24         imaname = os.path.split(imgpath)[1]  # 分離檔案路徑和檔名後獲取檔名（包括了字尾名）
25         # print(imaname)
26         img = cv2.imread(imgpath, cv2.IMREAD_COLOR)
27         cv2.namedWindow(imaname, cv2.WINDOW_AUTOSIZE)
28         cv2.imshow(imaname, img)
 
29         print(img.shape)
30     cv2.waitKey(0)
31 
32     return imglist
33 
34 if __name__ == '__main__':
35     imglist = read_img()
36     print(imglist.__len__())

最後圖片顯示