以下為大家分享一篇Python 讀取指定資料夾下的所有影像方法，具有很好的參考價值，希望對大家有幫助。一起來看看吧
（1）資料準備
資料集介紹：
資料集中存放的是1223幅圖像，其中756個負樣本（圖像名稱為0.1~0.756），458個正樣本（圖像名稱為1.1~1.458），其中："."前的標號為樣本標籤，"."後的標號為樣本序號
（2）利用python讀取資料夾中所有圖片
'''
Load the image files form the folder
input:
imgDir: the direction of the folder
imgName:the name of the folder
output:
data:the data of the dataset
label:the label of the datset
'''
def load_Img(imgDir,imgFoldName):
imgs = os.listdir(imgDir+imgFoldName)
imgNum = len(imgs)
data = np.empty((imgNum,1,12,12),dtype="float32")
label = np.empty((imgNum,),dtype="uint8")
for i in range (imgNum):
img = Image.open(imgDir+imgFoldName+"/"+imgs[i])
arr = np.asarray(img,dtype="float32")
data[i,:,:,:] = arr
label[i] = int(imgs[i].split('.')[0])
return data,label
##這裡得到的data和label都是ndarray資料data: (1223,1,12,12)
label:(1223,)craterDir = "./data/CraterImg/Adjust/" foldName = "East_CraterAdjust12" data, label = load_Img(craterDir,foldName)
相關推薦：
###### python讀取csv檔案並把檔案放入一個list中的實例講解############python實作對檔案中圖片產生標籤的txt檔案方法######### ################以上是Python 讀取指定資料夾下的所有映像方法的詳細內容。更多資訊請關注PHP中文網其他相關文章！
