python 讀取文件夾下的圖片並顯示

項目下的image文件夾里有五張.bmp圖像

import os
import cv2

def get_img_list(dir, firelist, ext=None):
    newdir = dir
    if os.path.isfile(dir):  # 如果是文件
        if ext is None:
            firelist.append(dir)
        elif ext in dir[-3:]:
            firelist.append(dir)
    elif os.path.isdir(dir):  # 如果是目錄
        for s in os.listdir(dir):
            newdir = os.path.join(dir, s)
            get_img_list(newdir, firelist, ext)

    return firelist

def read_img():
    image_path = './image'
    imglist = get_img_list(image_path, [], 'bmp')
    imgall = []
    for imgpath in imglist:
        # print(imgpath)
        imaname = os.path.split(imgpath)[1]  # 分離文件路徑和文件名后獲取文件名（包括了后綴名）
        # print(imaname)
        img = cv2.imread(imgpath, cv2.IMREAD_COLOR)
        imgall.append(img)
        cv2.namedWindow(imaname, cv2.WINDOW_AUTOSIZE)
        cv2.imshow(imaname, img)
        print(imaname, img.shape)
    cv2.waitKey(0)

    return imgall

if __name__ == '__main__':
    imgall = read_img()
    print(imgall.__len__())

 

最后圖片顯示