新聞中心
這里有您想知道的互聯(lián)網(wǎng)營銷解決方案
解決keras讀取多個(gè)hdf5文件進(jìn)行訓(xùn)練的方法-創(chuàng)新互聯(lián)
創(chuàng)新互聯(lián)www.cdcxhl.cn八線動(dòng)態(tài)BGP香港云服務(wù)器提供商,新人活動(dòng)買多久送多久,劃算不套路!
這篇文章將為大家詳細(xì)講解有關(guān)解決keras讀取多個(gè)hdf5文件進(jìn)行訓(xùn)練的方法,小編覺得挺實(shí)用的,因此分享給大家做個(gè)參考,希望大家閱讀完這篇文章后可以有所收獲。
用keras進(jìn)行大數(shù)據(jù)訓(xùn)練,為了加快訓(xùn)練,需要提前制作訓(xùn)練集。
由于HDF5的特性,所有數(shù)據(jù)需要一次性讀入到內(nèi)存中,才能保存。
為此,我采用分批次分為2個(gè)以上HDF5進(jìn)行存儲(chǔ)。
1、先讀取每個(gè)標(biāo)簽下的圖片,并設(shè)置標(biāo)簽
def load_dataset(path_name,data_path): images = [] labels = [] train_images = [] valid_images = [] train_labels = [] valid_labels = [] counter = 0 allpath = os.listdir(path_name) nb_classes = len(allpath) print("label_num: ",nb_classes) for child_dir in allpath: child_path = os.path.join(path_name, child_dir) for dir_image in os.listdir(child_path): if dir_image.endswith('.jpg'): img = cv2.imread(os.path.join(child_path, dir_image)) image = misc.imresize(img, (IMAGE_SIZE, IMAGE_SIZE), interp='bilinear') #resized_img = cv2.resize(img, (IMAGE_SIZE, IMAGE_SIZE)) images.append(image) labels.append(counter)
新聞名稱:解決keras讀取多個(gè)hdf5文件進(jìn)行訓(xùn)練的方法-創(chuàng)新互聯(lián)
鏈接地址:http://fisionsoft.com.cn/article/cseojo.html