keras讀寫檔案

讀寫檔案，主要程式碼函式

import os

image_types = (".jpg",".jpeg",".png",".bmp",".tif",".tiff")

def list_images(basePath,contains=None):
    #返回有效的檔案集
    retrun list_files(basePath,validExts = image_types,contains = contains)

def list_files(basePath,validExts = None,contains = None):
    #遍歷目錄結構
    for (rootDir,dirNames,filenames) in os.walk(basePath):
        #迴圈遍歷當前目錄中的檔名
        for filename in filenames:
            #如果contains字串不是none並且檔名不包含提供的字串，然後忽略檔案
            if contains is not None and filename.find(contains) == -1:#查詢檔名包含指定字串的檔案，不等於-1表明包含此字串。若是=-1，則表明不包含字串
                continue

            #確定當前檔案的副檔名
            ext = filename[filename.rfind("."):].lower()#將檔案字尾轉換成小寫

            #檢查檔案是否為影像，是否應進行處理
            if validExts is None or ext.endswith(validExts):
                #構造影像的路徑併產生它
                imagePath = os.path.join(rootDir,filename)#路徑拼接檔案路徑
                yield imagePath

使用程式碼語句

imagePaths = sorted(list(utils_paths.list_images(args["dataset"])))