Python 讀取指定資料夾下的所有映像方法

以下為大家分享一篇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,)

附註：nddary資料型別是numpy提供的資料型別，即N-dimensional array，它彌補了python中array不支援多維的缺陷

（3）呼叫方式

craterDir = "./data/CraterImg/Adjust/"
foldName = "East_CraterAdjust12"
data, label = load_Img(craterDir,foldName)

相關推薦：

###### python讀取csv檔案並把檔案放入一個list中的實例講解############python實作對檔案中圖片產生標籤的txt檔案方法######### ################

以上是Python 讀取指定資料夾下的所有映像方法的詳細內容。更多資訊請關注PHP中文網其他相關文章！