tensorflow - ModelCheckpoint 不保存 hdf5 文件
问题描述
我在构建语音到文本模型时遇到了一个奇怪的问题。
模型经过训练和创建,但是当我尝试将其保存为 hdf5 文件时,为了将其重用于测试目的,操作根本不起作用,并且控制台中没有弹出错误......</p>
这是完整的代码:
import os
import librosa
import IPython.display as ipd
import matplotlib.pyplot as plt
import numpy as np
from scipy.io import wavfile
import warnings
from sklearn.preprocessing import LabelEncoder
from keras.utils import np_utils
from sklearn.model_selection import train_test_split
from keras.layers import Dense, Dropout, Flatten, Conv1D, Input, MaxPooling1D
from keras.models import Model
from keras.callbacks import EarlyStopping, ModelCheckpoint
from keras import backend as K
from keras.models import load_model
import random
warnings.filterwarnings("ignore")
samples, sample_rate = librosa.load(r'C:\Users\makra\OneDrive\Desktop\Conda\Speech-to-text\train\audio\yes\0a7c2a8d_nohash_0.wav', sr = 16000)
ipd.Audio(samples, rate=sample_rate)
samples = librosa.resample(samples, sample_rate, 8000)
ipd.Audio(samples, rate=8000)
labels=os.listdir(r'C:\Users\makra\OneDrive\Desktop\Conda\Speech-to-text\train\audio')
labels=["yes", "no", "one", "two", "three", "four", "five", "six", "seven", "eight", "nine", "zero"]
all_wave = []
all_label = []
for label in labels:
print(label)
waves = [f for f in os.listdir(r'C:\Users\makra\OneDrive\Desktop\Conda\Speech-to-text\train\audio' + '/'+ label) if f.endswith('.wav')]
for wav in waves:
samples, sample_rate = librosa.load(r'C:\Users\makra\OneDrive\Desktop\Conda\Speech-to-text\train\audio' + '/' + label + '/' + wav, sr = 16000)
samples = librosa.resample(samples, sample_rate, 8000)
if(len(samples)== 8000) :
all_wave.append(samples)
all_label.append(label)
le = LabelEncoder()
y=le.fit_transform(all_label)
classes= list(le.classes_)
y=np_utils.to_categorical(y, num_classes=len(labels))
all_wave = np.array(all_wave).reshape(-1,8000,1)
x_tr, x_val, y_tr, y_val = train_test_split(np.array(all_wave),np.array(y),stratify=y,test_size = 0.2,random_state=777,shuffle=True)
K.clear_session()
inputs = Input(shape=(8000,1))
#First Conv1D layer
conv = Conv1D(8,13, padding='valid', activation='relu', strides=1)(inputs)
conv = MaxPooling1D(3)(conv)
conv = Dropout(0.3)(conv)
#Second Conv1D layer
conv = Conv1D(16, 11, padding='valid', activation='relu', strides=1)(conv)
conv = MaxPooling1D(3)(conv)
conv = Dropout(0.3)(conv)
#Third Conv1D layer
conv = Conv1D(32, 9, padding='valid', activation='relu', strides=1)(conv)
conv = MaxPooling1D(3)(conv)
conv = Dropout(0.3)(conv)
#Fourth Conv1D layer
conv = Conv1D(64, 7, padding='valid', activation='relu', strides=1)(conv)
conv = MaxPooling1D(3)(conv)
conv = Dropout(0.3)(conv)
#Flatten layer
conv = Flatten()(conv)
#Dense Layer 1
conv = Dense(256, activation='relu')(conv)
conv = Dropout(0.3)(conv)
#Dense Layer 2
conv = Dense(128, activation='relu')(conv)
conv = Dropout(0.3)(conv)
outputs = Dense(len(labels), activation='softmax')(conv)
model = Model(inputs, outputs)
model.summary()
model.compile(loss='categorical_crossentropy',optimizer='adam',metrics=['accuracy'])
es = EarlyStopping(monitor='val_loss', mode='min', verbose=1, patience=10, min_delta=0.0001)
mc = ModelCheckpoint('best_model.hdf5', monitor='val_acc', verbose=1, save_best_only=True, mode='max')
history=model.fit(x_tr, y_tr ,epochs=100, callbacks=[es,mc], batch_size=32, validation_data=(x_val,y_val))
非常感谢您的帮助,
马克拉姆
解决方案
我可以推荐save_weights_only=True
在您的回调中使用选项ModelCheckpoint
并使用 API model.load_weights(checkpoint_path)
。这将加载最新的模型权重,如此处所示。这只是一个临时解决方案,我认为您的模型精度没有增加,这就是您的save_best_only
选项无法保存任何模型的原因,因为您已经设置了精度应该提高的条件ModelCheckpoint
打回来。
推荐阅读
- javascript - 重定向后无法访问组件中的 this.props.location
- java - java在哪里寻找netbeans中的文件?是否可以更改目录?
- bots - Telegram BOT 跟踪谁添加了用户
- react-native - 如何将对讲聊天集成到 react-naive 移动应用程序?
- python - 如何使用数据框计算每小时的某些值
- r - 稀疏矩阵的 R repmat 函数
- android - Android 依赖 com.google.firebase:firebase-core 编译(12.0.1)和运行时(11.4.2)的不同版本
- c# - LTV 认证签名后,PDF 显示“文档已更改”
- ios - iOS 中的键盘输入速度非常慢
- c# - 使用 OWIN 自托管 WebApi 配置 IdentityServer4