python 讀取文件夾下的圖片進行處理

 

python的os模塊中有一個listdir函數可以遍歷讀取文件夾下的文件。

1 import os
2 
3 for filename in os.listdir(r"./file"):              #listdir的參數是文件夾的路徑
4     print ( filename)                                  #此時的filename是文件夾中文件的名稱
5     

利用上述代碼可以讀取文件夾下的所有文件，也包括圖片。

 

找到圖片以後，我們利用opencv中的imread函數讀取圖片，將其存儲爲一個numpy.narray類型的數組

import cv2                   #導入opencv庫
img1 = cv2.imread(filenames,cv2.IMREAD_GRAYSCALE)   #讀取圖片，第二個參數表示以灰度圖像讀入
if img1 is None:                   #判斷讀入的img1是否爲空，爲空就繼續下一輪循環
            continue
res1= cv2.resize(img1,(28,28))              #對圖片進行縮放，第一個參數是讀入的圖片，第二個是制定的縮放大小
res1_1 = res1.reshape(1,784)/255       #將表示圖片的二維矩陣轉換成一維
res1_1_1 = res1_1.tolist()                     #將numpy.narray類型的矩陣轉換成list
train_set_x.append(res1_1_1)               #將list添加到已有的list中